Searched refs:ShapeMap (Results 1 – 1 of 1) sorted by relevance
206 DenseMap<Value *, ShapeInfo> ShapeMap; member in __anon9f1638ae0111::LowerMatrixIntrinsics267 auto SIter = ShapeMap.find(V); in setShapeInfo()268 if (SIter != ShapeMap.end()) { in setShapeInfo()275 ShapeMap.insert({V, Shape}); in setShapeInfo()360 auto OpShape = ShapeMap.find(MatrixA); in propagateShapeForward()361 if (OpShape != ShapeMap.end()) in propagateShapeForward()367 auto OpShape = ShapeMap.find(Op.get()); in propagateShapeForward()368 if (OpShape != ShapeMap.end()) { in propagateShapeForward()378 if (ShapeMap.count(User) == 0) in propagateShapeForward()443 ShapeInfo Shape = ShapeMap[V]; in propagateShapeBackward()[all …]