Lines Matching defs:MatrixTy
343 class MatrixTy {
351 MatrixTy() : IsColumnMajor(MatrixLayout == MatrixLayoutTy::ColumnMajor) {}
352 MatrixTy(ArrayRef<Value *> Vectors)
355 MatrixTy(unsigned NumRows, unsigned NumColumns, Type *EltTy)
427 MatrixTy &addNumLoads(unsigned N) {
434 MatrixTy &addNumStores(unsigned N) {
439 MatrixTy &addNumExposedTransposes(unsigned N) {
444 MatrixTy &addNumComputeOps(unsigned N) {
500 MapVector<Value *, MatrixTy> Inst2ColumnMatrix;
544 MatrixTy getMatrix(Value *MatrixVal, const ShapeInfo &SI,
558 MatrixTy &M = Found->second;
1151 MatrixTy loadMatrix(Type *Ty, Value *Ptr, MaybeAlign MAlign, Value *Stride,
1157 MatrixTy Result;
1174 MatrixTy loadMatrix(Value *MatrixPtr, MaybeAlign Align, bool IsVolatile,
1216 void storeMatrix(const MatrixTy &StoreVal, Value *MatrixPtr,
1232 MatrixTy storeMatrix(Type *Ty, MatrixTy StoreVal, Value *Ptr,
1249 return MatrixTy().addNumStores(getNumOps(StoreVal.getVectorTy()) *
1338 void finalizeLowering(Instruction *Inst, MatrixTy Matrix,
1552 void emitMatrixMultiply(MatrixTy &Result, const MatrixTy &A,
1553 const MatrixTy &B, IRBuilder<> &Builder, bool IsTiled,
1757 MatrixTy getZeroMatrix(Type *EltType, unsigned R, unsigned C) {
1758 MatrixTy Res;
1781 MatrixTy TileResult;
1798 MatrixTy A =
1801 MatrixTy B =
1852 MatrixTy Res = getZeroMatrix(EltType, TileR, TileC);
1856 MatrixTy A =
1860 MatrixTy B =
1917 MatrixTy MA;
1918 MatrixTy MB;
1932 MatrixTy Result(R, C, EltType);
1943 Inst2ColumnMatrix[Transpose] = MatrixTy(M, C, EltType);
2045 const MatrixTy &Lhs = getMatrix(MatMul->getArgOperand(0), LShape, Builder);
2046 const MatrixTy &Rhs = getMatrix(MatMul->getArgOperand(1), RShape, Builder);
2055 MatrixTy Result(R, C, EltType);
2066 MatrixTy Result;
2071 MatrixTy InputMatrix = getMatrix(InputVal, ArgShape, Builder);
2138 MatrixTy Result;
2139 MatrixTy A = getMatrix(Lhs, Shape, Builder);
2140 MatrixTy B = getMatrix(Rhs, Shape, Builder);
2188 MatrixTy Result;
2189 MatrixTy M = getMatrix(Op, Shape, Builder);
2225 const MapVector<Value *, MatrixTy> &Inst2Matrix;
2242 const MapVector<Value *, MatrixTy> &Inst2Matrix,
2483 const MapVector<Value *, MatrixTy> &Inst2Matrix;
2488 RemarkGenerator(const MapVector<Value *, MatrixTy> &Inst2Matrix,