Lines Matching refs:ShapeInfo

350   struct ShapeInfo {  struct in __anon8ba1aee70111::LowerMatrixIntrinsics
356 ShapeInfo(unsigned NumRows = 0, unsigned NumColumns = 0) in ShapeInfo() function
360 ShapeInfo(Value *NumRows, Value *NumColumns) in ShapeInfo() function
361 : ShapeInfo(cast<ConstantInt>(NumRows)->getZExtValue(), in ShapeInfo()
364 bool operator==(const ShapeInfo &other) { in operator ==() argument
367 bool operator!=(const ShapeInfo &other) { return !(*this == other); } in operator !=() argument
397 ValueMap<Value *, ShapeInfo> ShapeMap;
451 MatrixTy getMatrix(Value *MatrixVal, const ShapeInfo &SI, in getMatrix()
490 bool setShapeInfo(Value *V, ShapeInfo Shape) { in setShapeInfo()
670 ShapeInfo Shape = ShapeMap[V]; in propagateShapeBackward()
974 bool IsVolatile, ShapeInfo Shape, IRBuilder<> &Builder) { in loadMatrix()
997 ShapeInfo MatrixShape, Value *I, Value *J, in loadMatrix()
998 ShapeInfo ResultShape, Type *EltTy, in loadMatrix()
1021 bool IsVolatile, ShapeInfo Shape) { in LowerLoad()
1045 MaybeAlign MAlign, bool IsVolatile, ShapeInfo MatrixShape, in storeMatrix()
1089 Value *Stride, bool IsVolatile, ShapeInfo Shape) { in LowerStore()
1371 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in isFusionProfitable()
1372 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in isFusionProfitable()
1411 void createTiledLoops(CallInst *MatMul, Value *LPtr, ShapeInfo LShape, in createTiledLoops()
1412 Value *RPtr, ShapeInfo RShape, StoreInst *Store) { in createTiledLoops()
1478 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in emitSIMDTiling()
1479 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in emitSIMDTiling()
1555 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in LowerMatrixMultiplyFused()
1556 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in LowerMatrixMultiplyFused()
1566 MA = getMatrix(A, ShapeInfo(R, M), Builder); in LowerMatrixMultiplyFused()
1567 MB = getMatrix(T, ShapeInfo(C, M), Builder); in LowerMatrixMultiplyFused()
1570 MA = getMatrix(T, ShapeInfo(R, M), Builder); in LowerMatrixMultiplyFused()
1571 MB = getMatrix(B, ShapeInfo(C, M), Builder); in LowerMatrixMultiplyFused()
1637 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in LowerMultiply()
1638 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in LowerMultiply()
1665 ShapeInfo ArgShape(Inst->getArgOperand(1), Inst->getArgOperand(2)); in LowerTranspose()
1731 ShapeInfo &Shape = I->second; in VisitBinaryOperator()
1781 ShapeInfo &Shape = I->second; in VisitUnaryOperator()