Lines Matching refs:ShapeMap
262 const ValueMap<Value *, ShapeInfo> &ShapeMap) { in computeShapeInfoForInst() argument
283 auto OpShape = ShapeMap.find(MatrixA); in computeShapeInfoForInst()
284 if (OpShape != ShapeMap.end()) in computeShapeInfoForInst()
291 auto OpShape = ShapeMap.find(Op.get()); in computeShapeInfoForInst()
292 if (OpShape != ShapeMap.end()) in computeShapeInfoForInst()
498 ValueMap<Value *, ShapeInfo> ShapeMap; member in __anon821fcdb70111::LowerMatrixIntrinsics
596 auto SIter = ShapeMap.find(V); in setShapeInfo()
597 if (SIter != ShapeMap.end()) { in setShapeInfo()
613 ShapeMap.insert({V, Shape}); in setShapeInfo()
656 if (auto SI = computeShapeInfoForInst(Inst, ShapeMap)) in propagateShapeForward()
662 if (ShapeMap.count(User) == 0) in propagateShapeForward()
726 ShapeInfo Shape = ShapeMap[V]; in propagateShapeBackward()
766 auto S = ShapeMap.find(&Old); in updateShapeAndReplaceAllUsesWith()
767 if (S != ShapeMap.end()) { in updateShapeAndReplaceAllUsesWith()
768 ShapeMap.erase(S); in updateShapeAndReplaceAllUsesWith()
770 ShapeMap.insert({New, S->second}); in updateShapeAndReplaceAllUsesWith()
917 assert(computeShapeInfoForInst(NewInst, ShapeMap) == in liftTranspose()
918 computeShapeInfoForInst(&I, ShapeMap) && in liftTranspose()
921 assert(computeShapeInfoForInst(Add, ShapeMap).value_or(ShapeMap[Add]) == in liftTranspose()
922 ShapeMap[Add] && in liftTranspose()
1003 if (ShapeMap.find(&I) == ShapeMap.end()) in Visit()
1321 if (ShapeMap.find(U.getUser()) == ShapeMap.end()) { in finalizeLowering()
1368 if (ShapeMap.find(Op) == ShapeMap.end()) in lowerDotProduct()
1387 if (match(Op, m_BinOp()) && ShapeMap.find(Op) != ShapeMap.end()) { in lowerDotProduct()
1466 if (match(Op, m_BinOp()) && ShapeMap.find(Op) != ShapeMap.end()) { in lowerDotProduct()
1467 ShapeMap[Op] = ShapeMap[Op].t(); in lowerDotProduct()
2075 auto I = ShapeMap.find(Inst); in VisitLoad()
2076 if (I == ShapeMap.end()) in VisitLoad()
2087 auto I = ShapeMap.find(StoredVal); in VisitStore()
2088 if (I == ShapeMap.end()) in VisitStore()
2099 auto I = ShapeMap.find(Inst); in VisitBinaryOperator()
2100 if (I == ShapeMap.end()) in VisitBinaryOperator()
2150 auto I = ShapeMap.find(Inst); in VisitUnaryOperator()
2151 if (I == ShapeMap.end()) in VisitUnaryOperator()