Searched refs:FindNodeByPrimitive (Results 1 – 2 of 2) sorted by relevance
215 std::vector<CNodePtr> FindNodeByPrimitive(const PrimitivePtr &primitive) const;216 std::vector<CNodePtr> FindNodeByPrimitive(const std::vector<PrimitivePtr> &primitive_list) const;
969 std::vector<CNodePtr> KernelGraph::FindNodeByPrimitive(const PrimitivePtr &primitive) const { in FindNodeByPrimitive() function in mindspore::session::KernelGraph980 std::vector<CNodePtr> KernelGraph::FindNodeByPrimitive(const std::vector<PrimitivePtr> &primitive_l… in FindNodeByPrimitive() function in mindspore::session::KernelGraph1299 auto call_nodes = FindNodeByPrimitive({std::make_shared<Primitive>(prim::kPrimCall->name()), in UpdateChildGraphOrder()