Lines Matching refs:ShapeMap
397 ValueMap<Value *, ShapeInfo> ShapeMap; member in __anon8ba1aee70111::LowerMatrixIntrinsics
495 auto SIter = ShapeMap.find(V); in setShapeInfo()
496 if (SIter != ShapeMap.end()) { in setShapeInfo()
503 ShapeMap.insert({V, Shape}); in setShapeInfo()
588 auto OpShape = ShapeMap.find(MatrixA); in propagateShapeForward()
589 if (OpShape != ShapeMap.end()) in propagateShapeForward()
595 auto OpShape = ShapeMap.find(Op.get()); in propagateShapeForward()
596 if (OpShape != ShapeMap.end()) { in propagateShapeForward()
606 if (ShapeMap.count(User) == 0) in propagateShapeForward()
670 ShapeInfo Shape = ShapeMap[V]; in propagateShapeBackward()
693 auto S = ShapeMap.find(&Old); in optimizeTransposes()
694 if (S != ShapeMap.end()) { in optimizeTransposes()
695 ShapeMap.erase(S); in optimizeTransposes()
697 ShapeMap.insert({New, S->second}); in optimizeTransposes()
850 if (ShapeMap.find(&I) == ShapeMap.end()) in Visit()
1182 if (ShapeMap.find(U.getUser()) == ShapeMap.end()) { in finalizeLowering()
1699 auto I = ShapeMap.find(Inst); in VisitLoad()
1700 if (I == ShapeMap.end()) in VisitLoad()
1711 auto I = ShapeMap.find(StoredVal); in VisitStore()
1712 if (I == ShapeMap.end()) in VisitStore()
1723 auto I = ShapeMap.find(Inst); in VisitBinaryOperator()
1724 if (I == ShapeMap.end()) in VisitBinaryOperator()
1774 auto I = ShapeMap.find(Inst); in VisitUnaryOperator()
1775 if (I == ShapeMap.end()) in VisitUnaryOperator()