Searched refs:prim_eval_implement_map (Results 1 – 3 of 3) sorted by relevance
153 static auto &prim_eval_implement_map = abstract::GetPrimitiveToEvalImplMap(); in InferPrimitiveShape() local154 auto ret = prim_eval_implement_map.find(prim); in InferPrimitiveShape()155 if (ret != prim_eval_implement_map.end()) { in InferPrimitiveShape()
91 static PrimitiveEvalImplMap prim_eval_implement_map = { in GetPrimitiveToEvalImplMap() local206 return prim_eval_implement_map; in GetPrimitiveToEvalImplMap()
940 auto &prim_eval_implement_map = abstract::GetPrimitiveToEvalImplMap(); in CppInferShape() local941 auto ret = prim_eval_implement_map.find(prim); in CppInferShape()942 if (ret != prim_eval_implement_map.end()) { in CppInferShape()