Searched refs:inputLHSTensorShape (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BatchMatmul.cpp | 203 Shape inputLHSTensorShape = context->getInputShape(kInputLHSTensor); in prepare() local
206 NN_RET_CHECK_EQ(getNumberOfDimensions(inputLHSTensorShape), in prepare()
209 NN_RET_CHECK_GE(getNumberOfDimensions(inputLHSTensorShape), 2u) in prepare()
211 NN_RET_CHECK_LE(getNumberOfDimensions(inputLHSTensorShape), 4u) in prepare()
213 uint32_t numDims = getNumberOfDimensions(inputLHSTensorShape); in prepare()
217 NN_RET_CHECK(canMatrixMul(getSizeOfDimension(inputLHSTensorShape, numDims - 2), in prepare()
218 getSizeOfDimension(inputLHSTensorShape, numDims - 1), in prepare()
225 computeOutputDimensions(inputLHSTensorShape, inputRHSTensorShape, adjX, adjY); in prepare()
|