Lines Matching defs:MatrixTy

355   class MatrixTy {
363 MatrixTy() : IsColumnMajor(MatrixLayout == MatrixLayoutTy::ColumnMajor) {}
364 MatrixTy(ArrayRef<Value *> Vectors)
367 MatrixTy(unsigned NumRows, unsigned NumColumns, Type *EltTy)
439 MatrixTy &addNumLoads(unsigned N) {
446 MatrixTy &addNumStores(unsigned N) {
451 MatrixTy &addNumExposedTransposes(unsigned N) {
456 MatrixTy &addNumComputeOps(unsigned N) {
506 MapVector<Value *, MatrixTy> Inst2ColumnMatrix;
552 MatrixTy getMatrix(Value *MatrixVal, const ShapeInfo &SI,
566 MatrixTy &M = Found->second;
1124 MatrixTy loadMatrix(Type *Ty, Value *Ptr, MaybeAlign MAlign, Value *Stride,
1130 MatrixTy Result;
1147 MatrixTy loadMatrix(Value *MatrixPtr, MaybeAlign Align, bool IsVolatile,
1189 void storeMatrix(const MatrixTy &StoreVal, Value *MatrixPtr,
1205 MatrixTy storeMatrix(Type *Ty, MatrixTy StoreVal, Value *Ptr,
1222 return MatrixTy().addNumStores(getNumOps(StoreVal.getVectorTy()) *
1312 void finalizeLowering(Instruction *Inst, MatrixTy Matrix,
1523 void emitMatrixMultiply(MatrixTy &Result, const MatrixTy &A,
1524 const MatrixTy &B, IRBuilder<> &Builder, bool IsTiled,
1728 MatrixTy getZeroMatrix(Type *EltType, unsigned R, unsigned C) {
1729 MatrixTy Res;
1752 MatrixTy TileResult;
1769 MatrixTy A =
1772 MatrixTy B =
1823 MatrixTy Res = getZeroMatrix(EltType, TileR, TileC);
1827 MatrixTy A =
1831 MatrixTy B =
1888 MatrixTy MA;
1889 MatrixTy MB;
1903 MatrixTy Result(R, C, EltType);
1914 Inst2ColumnMatrix[Transpose] = MatrixTy(M, C, EltType);
2016 const MatrixTy &Lhs = getMatrix(MatMul->getArgOperand(0), LShape, Builder);
2017 const MatrixTy &Rhs = getMatrix(MatMul->getArgOperand(1), RShape, Builder);
2026 MatrixTy Result(R, C, EltType);
2037 MatrixTy Result;
2042 MatrixTy InputMatrix = getMatrix(InputVal, ArgShape, Builder);
2109 MatrixTy Result;
2110 MatrixTy A = getMatrix(Lhs, Shape, Builder);
2111 MatrixTy B = getMatrix(Rhs, Shape, Builder);
2159 MatrixTy Result;
2160 MatrixTy M = getMatrix(Op, Shape, Builder);
2196 const MapVector<Value *, MatrixTy> &Inst2Matrix;
2213 const MapVector<Value *, MatrixTy> &Inst2Matrix,
2457 const MapVector<Value *, MatrixTy> &Inst2Matrix;
2462 RemarkGenerator(const MapVector<Value *, MatrixTy> &Inst2Matrix,