Searched refs:ShapeMap (Results 1 – 1 of 1) sorted by relevance
262 const ValueMap<Value *, ShapeInfo> &ShapeMap) { in computeShapeInfoForInst() argument283 auto OpShape = ShapeMap.find(MatrixA); in computeShapeInfoForInst()284 if (OpShape != ShapeMap.end()) in computeShapeInfoForInst()291 auto OpShape = ShapeMap.find(Op.get()); in computeShapeInfoForInst()292 if (OpShape != ShapeMap.end()) in computeShapeInfoForInst()498 ValueMap<Value *, ShapeInfo> ShapeMap; member in __anon821fcdb70111::LowerMatrixIntrinsics596 auto SIter = ShapeMap.find(V); in setShapeInfo()597 if (SIter != ShapeMap.end()) { in setShapeInfo()613 ShapeMap.insert({V, Shape}); in setShapeInfo()656 if (auto SI = computeShapeInfoForInst(Inst, ShapeMap)) in propagateShapeForward()[all …]