Searched refs:ShapeMap (Results 1 – 2 of 2) sorted by relevance
206 DenseMap<Value *, ShapeInfo> ShapeMap; member in __anon3a55b4310111::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 …]
438 ValueMap<Value *, ShapeInfo> ShapeMap; member in __anon71f215770111::LowerMatrixIntrinsics536 auto SIter = ShapeMap.find(V); in setShapeInfo()537 if (SIter != ShapeMap.end()) { in setShapeInfo()544 ShapeMap.insert({V, Shape}); in setShapeInfo()629 auto OpShape = ShapeMap.find(MatrixA); in propagateShapeForward()630 if (OpShape != ShapeMap.end()) in propagateShapeForward()636 auto OpShape = ShapeMap.find(Op.get()); in propagateShapeForward()637 if (OpShape != ShapeMap.end()) { in propagateShapeForward()647 if (ShapeMap.count(User) == 0) in propagateShapeForward()711 ShapeInfo Shape = ShapeMap[V]; in propagateShapeBackward()[all …]