Lines Matching refs:ShapeInfo
386 struct ShapeInfo { struct in __anon04e69bad0111::LowerMatrixIntrinsics
392 ShapeInfo(unsigned NumRows = 0, unsigned NumColumns = 0) in ShapeInfo() function
396 ShapeInfo(Value *NumRows, Value *NumColumns) in ShapeInfo() function
397 : ShapeInfo(cast<ConstantInt>(NumRows)->getZExtValue(), in ShapeInfo()
400 bool operator==(const ShapeInfo &other) { in operator ==() argument
403 bool operator!=(const ShapeInfo &other) { return !(*this == other); } in operator !=() argument
425 ShapeInfo t() const { return ShapeInfo(NumColumns, NumRows); } in t() argument
436 ValueMap<Value *, ShapeInfo> ShapeMap;
490 MatrixTy getMatrix(Value *MatrixVal, const ShapeInfo &SI, in getMatrix()
529 bool setShapeInfo(Value *V, ShapeInfo Shape) { in setShapeInfo()
718 ShapeInfo Shape = ShapeMap[V]; in propagateShapeBackward()
739 Value *Op0, ShapeInfo Shape0, Value *Op1, ShapeInfo Shape1, in distributeTransposes() argument
741 function_ref<Instruction *(Value *, ShapeInfo, Value *, ShapeInfo)> in distributeTransposes()
805 [&](Value *T0, ShapeInfo Shape0, Value *T1, ShapeInfo Shape1) { in sinkTranspose()
827 [&](Value *T0, ShapeInfo Shape0, Value *T1, ShapeInfo Shape1) { in sinkTranspose()
847 [&](Value *T0, ShapeInfo Shape0, Value *T1, ShapeInfo Shape1) { in sinkTranspose()
1106 bool IsVolatile, ShapeInfo Shape, IRBuilder<> &Builder) { in loadMatrix()
1129 ShapeInfo MatrixShape, Value *I, Value *J, in loadMatrix()
1130 ShapeInfo ResultShape, Type *EltTy, in loadMatrix()
1147 bool IsVolatile, ShapeInfo Shape) { in LowerLoad()
1171 MaybeAlign MAlign, bool IsVolatile, ShapeInfo MatrixShape, in storeMatrix()
1209 Value *Stride, bool IsVolatile, ShapeInfo Shape) { in LowerStore()
1319 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in lowerDotProduct()
1320 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in lowerDotProduct()
1649 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in isFusionProfitable()
1650 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in isFusionProfitable()
1689 void createTiledLoops(CallInst *MatMul, Value *LPtr, ShapeInfo LShape, in createTiledLoops()
1690 Value *RPtr, ShapeInfo RShape, StoreInst *Store) { in createTiledLoops()
1756 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in emitSIMDTiling()
1757 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in emitSIMDTiling()
1833 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in LowerMatrixMultiplyFused()
1834 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in LowerMatrixMultiplyFused()
1844 MA = getMatrix(A, ShapeInfo(R, M), Builder); in LowerMatrixMultiplyFused()
1845 MB = getMatrix(T, ShapeInfo(C, M), Builder); in LowerMatrixMultiplyFused()
1848 MA = getMatrix(T, ShapeInfo(R, M), Builder); in LowerMatrixMultiplyFused()
1849 MB = getMatrix(B, ShapeInfo(C, M), Builder); in LowerMatrixMultiplyFused()
1915 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in LowerMultiply()
1916 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in LowerMultiply()
1943 ShapeInfo ArgShape(Inst->getArgOperand(1), Inst->getArgOperand(2)); in LowerTranspose()
2009 ShapeInfo &Shape = I->second; in VisitBinaryOperator()
2059 ShapeInfo &Shape = I->second; in VisitUnaryOperator()