44 std::array<int64_t, 2> getTileShape(ArrayRef<int64_t> operandShape,  in getTileShape()  argument
47   return {operandShape[0] / kNumRowsPerTile,  in getTileShape()
48           (operandShape[1] * elementType.getIntOrFloatBitWidth()) /  in getTileShape()
156                                                  ArrayRef<int64_t> operandShape,  in getRegisterIndexToTileOffsetMap()  argument
163       getTileShape(operandShape, elementType, lineSize);  in getRegisterIndexToTileOffsetMap()
176   ArrayRef<int64_t> operandShape = fragmentType.vectorType.getShape();  in getLaneIdAndValueIdToOperandCoord()  local
193       lineSize, elementType, operandShape,  in getLaneIdAndValueIdToOperandCoord()
245   ArrayRef<int64_t> operandShape = params.fragmentType.getShape();  in getLaneIdToLdMatrixMatrixCoord()  local
254     AffineExpr row = d0 % (operandShape[0]);  in getLaneIdToLdMatrixMatrixCoord()
255     AffineExpr col = d0.floorDiv(operandShape[0]) * (kElementsPer128b);  in getLaneIdToLdMatrixMatrixCoord()
262     const int64_t num8x128bCols = (operandShape[0] * bitsPerElement) / 128;  in getLaneIdToLdMatrixMatrixCoord()