Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DBatchMatmul.cpp204 Shape inputRHSTensorShape = context->getInputShape(kInputRHSTensor); in prepare() local
207 getNumberOfDimensions(inputRHSTensorShape)) in prepare()
219 getSizeOfDimension(inputRHSTensorShape, numDims - 2), in prepare()
220 getSizeOfDimension(inputRHSTensorShape, numDims - 1), adjX, adjY)) in prepare()
225 computeOutputDimensions(inputLHSTensorShape, inputRHSTensorShape, adjX, adjY); in prepare()