Lines Matching refs:adjY
49 bool adjY) { in canMatrixMul() argument
56 if (adjY) { in canMatrixMul()
64 const Shape& RHSTensorShape, bool adjX, bool adjY) { in computeOutputDimensions() argument
70 adjY ? RHSTensorShape.dimensions[numDims - 2] : RHSTensorShape.dimensions[numDims - 1]; in computeOutputDimensions()
117 const Shape& inputRHSShape, const bool adjX, const bool adjY, T* outputData, in batchMatMulGeneric() argument
137 if (!adjY) { in batchMatMulGeneric()
142 Shape realInputRHSShape = adjY ? inputRHSShape : swapRowColumnDims(inputRHSShape); in batchMatMulGeneric()
153 const Shape& inputRHSShape, const bool adjX, const bool adjY, in batchMatMulQuantized() argument
165 if (!adjY) { in batchMatMulQuantized()
170 Shape realInputRHSShape = adjY ? inputRHSShape : swapRowColumnDims(inputRHSShape); in batchMatMulQuantized()
215 const bool adjY = context->getInputValue<bool>(kInputRHSAdj); in prepare() local
220 getSizeOfDimension(inputRHSTensorShape, numDims - 1), adjX, adjY)) in prepare()
225 computeOutputDimensions(inputLHSTensorShape, inputRHSTensorShape, adjX, adjY); in prepare()