Lines Matching refs:ShapeMap
396 ValueMap<Value *, ShapeInfo> ShapeMap; member in __anonf095cceb0111::LowerMatrixIntrinsics
494 auto SIter = ShapeMap.find(V); in setShapeInfo()
495 if (SIter != ShapeMap.end()) { in setShapeInfo()
502 ShapeMap.insert({V, Shape}); in setShapeInfo()
587 auto OpShape = ShapeMap.find(MatrixA); in propagateShapeForward()
588 if (OpShape != ShapeMap.end()) in propagateShapeForward()
594 auto OpShape = ShapeMap.find(Op.get()); in propagateShapeForward()
595 if (OpShape != ShapeMap.end()) { in propagateShapeForward()
605 if (ShapeMap.count(User) == 0) in propagateShapeForward()
669 ShapeInfo Shape = ShapeMap[V]; in propagateShapeBackward()
692 auto S = ShapeMap.find(&Old); in optimizeTransposes()
693 if (S != ShapeMap.end()) { in optimizeTransposes()
694 ShapeMap.erase(S); in optimizeTransposes()
696 ShapeMap.insert({New, S->second}); in optimizeTransposes()
852 if (ShapeMap.find(&I) == ShapeMap.end()) in Visit()
1182 if (ShapeMap.find(U.getUser()) == ShapeMap.end()) { in finalizeLowering()
1691 auto I = ShapeMap.find(Inst); in VisitLoad()
1692 if (I == ShapeMap.end()) in VisitLoad()
1703 auto I = ShapeMap.find(StoredVal); in VisitStore()
1704 if (I == ShapeMap.end()) in VisitStore()
1715 auto I = ShapeMap.find(Inst); in VisitBinaryOperator()
1716 if (I == ShapeMap.end()) in VisitBinaryOperator()
1766 auto I = ShapeMap.find(Inst); in VisitUnaryOperator()
1767 if (I == ShapeMap.end()) in VisitUnaryOperator()