Lines Matching refs:NumColumns
263 MatrixTy(unsigned NumRows, unsigned NumColumns, Type *EltTy) in MatrixTy() argument
266 unsigned D = isColumnMajor() ? NumColumns : NumRows; in MatrixTy()
269 EltTy, isColumnMajor() ? NumRows : NumColumns))); in MatrixTy()
388 unsigned NumColumns; member
392 ShapeInfo(unsigned NumRows = 0, unsigned NumColumns = 0) in ShapeInfo()
393 : NumRows(NumRows), NumColumns(NumColumns), in ShapeInfo()
396 ShapeInfo(Value *NumRows, Value *NumColumns) in ShapeInfo()
398 cast<ConstantInt>(NumColumns)->getZExtValue()) {} in ShapeInfo()
401 return NumRows == other.NumRows && NumColumns == other.NumColumns; in operator ==()
408 assert(NumRows == 0 || NumColumns != 0); in operator bool()
415 return NumColumns; in getStride()
420 return NumColumns; in getNumVectors()
425 ShapeInfo t() const { return ShapeInfo(NumColumns, NumRows); } in t()
495 SI.NumRows * SI.NumColumns && in getMatrix()
507 if (SI.NumRows == M.getNumRows() && SI.NumColumns == M.getNumColumns()) in getMatrix()
537 SIter->second.NumColumns != Shape.NumColumns)) { in setShapeInfo()
539 << SIter->second.NumColumns << " vs " << Shape.NumRows << "x" in setShapeInfo()
540 << Shape.NumColumns << ") for " << *V << "\n"; in setShapeInfo()
547 << SIter->second.NumColumns << " for " << *V << "\n"); in setShapeInfo()
552 LLVM_DEBUG(dbgs() << " " << Shape.NumRows << " x " << Shape.NumColumns in setShapeInfo()
744 Op0, Shape0.NumRows, Shape0.NumColumns, Op0->getName() + "_t"); in distributeTransposes()
749 Op1, Shape1.NumRows, Shape1.NumColumns, Op1->getName() + "_t"); in distributeTransposes()
807 Shape0.NumColumns, in sinkTranspose()
808 Shape1.NumColumns, "mmul"); in sinkTranspose()
1138 ResultShape.NumColumns); in loadMatrix()
1322 if (LShape.NumRows != 1 || RShape.NumColumns != 1) // not a dot product in lowerDotProduct()
1394 auto LHSCost = GetCostForArg(LHS, LShape.NumColumns); in lowerDotProduct()
1406 (LShape.NumColumns - 1) + in lowerDotProduct()
1408 (LShape.NumColumns); in lowerDotProduct()
1653 const unsigned C = RShape.NumColumns; in isFusionProfitable()
1654 const unsigned M = LShape.NumColumns; in isFusionProfitable()
1694 TileInfo TI(LShape.NumRows, RShape.NumColumns, LShape.NumColumns, TileSize); in createTiledLoops()
1733 Store->isVolatile(), {LShape.NumRows, RShape.NumColumns}, in createTiledLoops()
1743 unsigned InnerLoopUnrollCount = std::min(10u, LShape.NumColumns / TileSize); in createTiledLoops()
1760 const unsigned C = RShape.NumColumns; in emitSIMDTiling()
1761 const unsigned M = LShape.NumColumns; in emitSIMDTiling()
1836 const unsigned M = LShape.NumColumns; in LowerMatrixMultiplyFused()
1837 const unsigned C = RShape.NumColumns; in LowerMatrixMultiplyFused()
1924 const unsigned C = RShape.NumColumns; in LowerMultiply()
1925 assert(LShape.NumColumns == RShape.NumRows); in LowerMultiply()
1947 InputMatrix.isColumnMajor() ? ArgShape.NumRows : ArgShape.NumColumns; in LowerTranspose()
1949 InputMatrix.isColumnMajor() ? ArgShape.NumColumns : ArgShape.NumRows; in LowerTranspose()
1970 Result.addNumComputeOps(2 * ArgShape.NumRows * ArgShape.NumColumns) in LowerTranspose()