Lines Matching defs:ShapeMap

249                         const DenseMap<Value *, ShapeInfo> &ShapeMap) {
270     auto OpShape = ShapeMap.find(MatrixA);
271     if (OpShape != ShapeMap.end())
278       auto OpShape = ShapeMap.find(Op.get());
279       if (OpShape != ShapeMap.end())
484   /// erasing or RAUW'ing a value that is present in ShapeMap. If the
487   /// ShapeMap.  We don't use ValueMap, as there are also cases where we do not
489   /// erasing a value with an entry in ShapeMap, use
490   /// eraseFromParentAndRemoveFromShapeMap to make sure ShapeMap is also updated
492   DenseMap<Value *, ShapeInfo> ShapeMap;
588     auto SIter = ShapeMap.find(V);
589     if (SIter != ShapeMap.end()) {
605     ShapeMap.insert({V, Shape});
648       if (auto SI = computeShapeInfoForInst(Inst, ShapeMap))
654           if (ShapeMap.count(User) == 0)
718         ShapeInfo Shape = ShapeMap[V];
754   /// Erase \p Inst from both ShapeMap (if an entry exists) and erase \p Inst
757     auto Iter = ShapeMap.find(Inst);
758     if (Iter != ShapeMap.end())
759       ShapeMap.erase(Iter);
776   /// Add a new entry to ShapeMap for \p New with \p Old's shape info, erase the
779     // We need to remove Old from the ShapeMap otherwise RAUW will replace it
782     auto S = ShapeMap.find(&Old);
783     if (S != ShapeMap.end()) {
784       ShapeMap.erase(S);
786         ShapeMap.insert({New, S->second});
932       assert(computeShapeInfoForInst(NewInst, ShapeMap) ==
933                  computeShapeInfoForInst(&I, ShapeMap) &&
939             computeShapeInfoForInst(AddI, ShapeMap).value_or(ShapeMap[AddI]) ==
940                 ShapeMap[AddI] &&
1029         if (ShapeMap.find(&I) == ShapeMap.end())
1347       if (ShapeMap.find(U.getUser()) == ShapeMap.end()) {
1394       if (ShapeMap.find(Op) == ShapeMap.end())
1413       if (match(Op, m_BinOp()) && ShapeMap.find(Op) != ShapeMap.end()) {
1493         auto It = ShapeMap.find(Op);
1494         if (It != ShapeMap.end()) {
2104     auto I = ShapeMap.find(Inst);
2105     if (I == ShapeMap.end())
2116     auto I = ShapeMap.find(StoredVal);
2117     if (I == ShapeMap.end())
2128     auto I = ShapeMap.find(Inst);
2129     if (I == ShapeMap.end())
2179     auto I = ShapeMap.find(Inst);
2180     if (I == ShapeMap.end())