Lines Matching refs:ShapeInfo
197 struct ShapeInfo { struct
203 ShapeInfo(unsigned NumRows = 0, unsigned NumColumns = 0) in ShapeInfo() function
207 ShapeInfo(Value *NumRows, Value *NumColumns) in ShapeInfo() function
208 : ShapeInfo(cast<ConstantInt>(NumRows)->getZExtValue(), in ShapeInfo()
211 bool operator==(const ShapeInfo &other) { in operator ==() argument
214 bool operator!=(const ShapeInfo &other) { return !(*this == other); } in operator !=() argument
236 ShapeInfo t() const { return ShapeInfo(NumColumns, NumRows); } in t() argument
260 static std::optional<ShapeInfo>
262 const ValueMap<Value *, ShapeInfo> &ShapeMap) { in computeShapeInfoForInst()
268 return ShapeInfo(M, K); in computeShapeInfoForInst()
272 return ShapeInfo(N, M); in computeShapeInfoForInst()
277 return ShapeInfo(N, M); in computeShapeInfoForInst()
280 return ShapeInfo(M, N); in computeShapeInfoForInst()
498 ValueMap<Value *, ShapeInfo> ShapeMap;
552 MatrixTy getMatrix(Value *MatrixVal, const ShapeInfo &SI, in getMatrix()
591 bool setShapeInfo(Value *V, ShapeInfo Shape) { in setShapeInfo()
726 ShapeInfo Shape = ShapeMap[V]; in propagateShapeBackward()
747 Value *Op0, ShapeInfo Shape0, Value *Op1, ShapeInfo Shape1, in distributeTransposes() argument
749 function_ref<Instruction *(Value *, ShapeInfo, Value *, ShapeInfo)> in distributeTransposes()
813 [&](Value *T0, ShapeInfo Shape0, Value *T1, ShapeInfo Shape1) { in sinkTranspose()
835 [&](Value *T0, ShapeInfo Shape0, Value *T1, ShapeInfo Shape1) { in sinkTranspose()
855 [&](Value *T0, ShapeInfo Shape0, Value *T1, ShapeInfo Shape1) { in sinkTranspose()
1125 bool IsVolatile, ShapeInfo Shape, IRBuilder<> &Builder) { in loadMatrix()
1148 ShapeInfo MatrixShape, Value *I, Value *J, in loadMatrix()
1149 ShapeInfo ResultShape, Type *EltTy, in loadMatrix()
1166 bool IsVolatile, ShapeInfo Shape) { in LowerLoad()
1190 MaybeAlign MAlign, bool IsVolatile, ShapeInfo MatrixShape, in storeMatrix()
1228 Value *Stride, bool IsVolatile, ShapeInfo Shape) { in LowerStore()
1338 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in lowerDotProduct()
1339 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in lowerDotProduct()
1696 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in isFusionProfitable()
1697 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in isFusionProfitable()
1736 void createTiledLoops(CallInst *MatMul, Value *LPtr, ShapeInfo LShape, in createTiledLoops()
1737 Value *RPtr, ShapeInfo RShape, StoreInst *Store) { in createTiledLoops()
1803 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in emitSIMDTiling()
1804 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in emitSIMDTiling()
1882 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in LowerMatrixMultiplyFused()
1883 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in LowerMatrixMultiplyFused()
1893 MA = getMatrix(A, ShapeInfo(R, M), Builder); in LowerMatrixMultiplyFused()
1894 MB = getMatrix(T, ShapeInfo(C, M), Builder); in LowerMatrixMultiplyFused()
1897 MA = getMatrix(T, ShapeInfo(R, M), Builder); in LowerMatrixMultiplyFused()
1898 MB = getMatrix(B, ShapeInfo(C, M), Builder); in LowerMatrixMultiplyFused()
2013 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in LowerMultiply()
2014 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in LowerMultiply()
2041 ShapeInfo ArgShape(Inst->getArgOperand(1), Inst->getArgOperand(2)); in LowerTranspose()
2107 ShapeInfo &Shape = I->second; in VisitBinaryOperator()
2157 ShapeInfo &Shape = I->second; in VisitUnaryOperator()