Searched refs:get_prim_graph (Results 1 – 1 of 1) sorted by relevance
421 …ction<std::shared_ptr<FuncGraph>(const PrimitivePtr, const AbstractFunctionPtr)> &get_prim_graph) { in TraverseGraphMap() argument448 FuncGraphPtr g = get_prim_graph(GetValueNode<PrimitivePtr>(const_primitive_node), in TraverseGraphMap()463 auto get_prim_graph = [&prim_graphs](const PrimitivePtr &prim, const AbstractFunctionPtr &type) { in WrapPrimitives() local492 TraverseGraphMap(manager_ptr, &tr, fgs, get_prim_graph); in WrapPrimitives()