Lines Matching refs:ShapeInfo
349 struct ShapeInfo { struct in __anonf095cceb0111::LowerMatrixIntrinsics
355 ShapeInfo(unsigned NumRows = 0, unsigned NumColumns = 0) in ShapeInfo() function
359 ShapeInfo(Value *NumRows, Value *NumColumns) in ShapeInfo() function
360 : ShapeInfo(cast<ConstantInt>(NumRows)->getZExtValue(), in ShapeInfo()
363 bool operator==(const ShapeInfo &other) { in operator ==() argument
366 bool operator!=(const ShapeInfo &other) { return !(*this == other); } in operator !=() argument
396 ValueMap<Value *, ShapeInfo> ShapeMap;
450 MatrixTy getMatrix(Value *MatrixVal, const ShapeInfo &SI, in getMatrix()
489 bool setShapeInfo(Value *V, ShapeInfo Shape) { in setShapeInfo()
669 ShapeInfo Shape = ShapeMap[V]; in propagateShapeBackward()
977 bool IsVolatile, ShapeInfo Shape, IRBuilder<> &Builder) { in loadMatrix()
999 ShapeInfo MatrixShape, Value *I, Value *J, in loadMatrix()
1000 ShapeInfo ResultShape, Type *EltTy, in loadMatrix()
1023 bool IsVolatile, ShapeInfo Shape) { in LowerLoad()
1047 MaybeAlign MAlign, bool IsVolatile, ShapeInfo MatrixShape, in storeMatrix()
1089 Value *Stride, bool IsVolatile, ShapeInfo Shape) { in LowerStore()
1366 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in isFusionProfitable()
1367 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in isFusionProfitable()
1405 void createTiledLoops(CallInst *MatMul, Value *LPtr, ShapeInfo LShape, in createTiledLoops()
1406 Value *RPtr, ShapeInfo RShape, StoreInst *Store) { in createTiledLoops()
1470 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in emitSIMDTiling()
1471 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in emitSIMDTiling()
1547 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in LowerMatrixMultiplyFused()
1548 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in LowerMatrixMultiplyFused()
1558 MA = getMatrix(A, ShapeInfo(R, M), Builder); in LowerMatrixMultiplyFused()
1559 MB = getMatrix(T, ShapeInfo(C, M), Builder); in LowerMatrixMultiplyFused()
1562 MA = getMatrix(T, ShapeInfo(R, M), Builder); in LowerMatrixMultiplyFused()
1563 MB = getMatrix(B, ShapeInfo(C, M), Builder); in LowerMatrixMultiplyFused()
1629 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in LowerMultiply()
1630 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in LowerMultiply()
1657 ShapeInfo ArgShape(Inst->getArgOperand(1), Inst->getArgOperand(2)); in LowerTranspose()
1723 ShapeInfo &Shape = I->second; in VisitBinaryOperator()
1773 ShapeInfo &Shape = I->second; in VisitUnaryOperator()