Searched refs:inputLHSShape (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BatchMatmul.cpp | 116 bool batchMatMulGeneric(const T* inputLHSData, const Shape& inputLHSShape, const T* inputRHSData, in batchMatMulGeneric() argument
125 auto tempInputLHSData = getTempData<T>(getNumberOfElements(inputLHSShape)); in batchMatMulGeneric()
134 transposeRowsColumns(inputLHSData, inputLHSShape, tempInputLHSData.get()); in batchMatMulGeneric()
141 Shape realInputLHSShape = adjX ? inputLHSShape : swapRowColumnDims(inputLHSShape); in batchMatMulGeneric()
152 bool batchMatMulQuantized(const T* inputLHSData, const Shape& inputLHSShape, const T* inputRHSData, in batchMatMulQuantized() argument
159 auto tempInputLHSData = getTempData<T>(getNumberOfElements(inputLHSShape)); in batchMatMulQuantized()
162 transposeRowsColumns(inputLHSData, inputLHSShape, tempInputLHSData.get()); in batchMatMulQuantized()
169 Shape realInputLHSShape = adjX ? inputLHSShape : swapRowColumnDims(inputLHSShape); in batchMatMulQuantized()
|