Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DBatchMatmul.cpp204 Shape inputRHSTensorShape = context->getInputShape(kInputRHSTensor); in prepare()
234 context->getInputBuffer<float>(kInputRHSTensor), in execute()
235 context->getInputShape(kInputRHSTensor), in execute()
243 context->getInputBuffer<_Float16>(kInputRHSTensor), in execute()
244 context->getInputShape(kInputRHSTensor), in execute()
252 context->getInputBuffer<int32_t>(kInputRHSTensor), in execute()
253 context->getInputShape(kInputRHSTensor), in execute()
261 context->getInputBuffer<int8_t>(kInputRHSTensor), in execute()
262 context->getInputShape(kInputRHSTensor), in execute()
/packages/modules/NeuralNetworks/common/types/operations/include/
DBatchMatmul.h28 constexpr uint32_t kInputRHSTensor = 1; variable
/packages/modules/NeuralNetworks/common/types/operations/src/
DBatchMatmul.cpp30 OperandType inputRHSTensorType = context->getInputType(kInputRHSTensor); in validate()