Home
last modified time | relevance | path

Searched refs:kInputLHSTensor (Results 1 – 3 of 3) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DBatchMatmul.cpp203 Shape inputLHSTensorShape = context->getInputShape(kInputLHSTensor); in prepare()
230 switch (context->getInputType(kInputLHSTensor)) { in execute()
232 return batchMatMulGeneric(context->getInputBuffer<float>(kInputLHSTensor), in execute()
233 context->getInputShape(kInputLHSTensor), in execute()
241 return batchMatMulGeneric(context->getInputBuffer<_Float16>(kInputLHSTensor), in execute()
242 context->getInputShape(kInputLHSTensor), in execute()
250 return batchMatMulGeneric(context->getInputBuffer<int32_t>(kInputLHSTensor), in execute()
251 context->getInputShape(kInputLHSTensor), in execute()
259 return batchMatMulQuantized(context->getInputBuffer<int8_t>(kInputLHSTensor), in execute()
260 context->getInputShape(kInputLHSTensor), in execute()
/packages/modules/NeuralNetworks/common/types/operations/include/
DBatchMatmul.h27 constexpr uint32_t kInputLHSTensor = 0; variable
/packages/modules/NeuralNetworks/common/types/operations/src/
DBatchMatmul.cpp29 OperandType inputLHSTensorType = context->getInputType(kInputLHSTensor); in validate()