Searched refs:adjY (Results 1 – 1 of 1) sorted by relevance
49 bool adjY) { in canMatrixMul() argument56 if (adjY) { in canMatrixMul()64 const Shape& RHSTensorShape, bool adjX, bool adjY) { in computeOutputDimensions() argument70 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() argument137 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() argument165 if (!adjY) { in batchMatMulQuantized()170 Shape realInputRHSShape = adjY ? inputRHSShape : swapRowColumnDims(inputRHSShape); in batchMatMulQuantized()[all …]