| /third_party/mindspore/mindspore/core/ir/ |
| D | func_graph.cc | 36 FuncGraph::FuncGraph() in FuncGraph() function in mindspore::FuncGraph
57 abstract::AbstractBasePtr FuncGraph::ToAbstract() { in ToAbstract()
59 …return std::make_shared<abstract::FuncGraphAbstractClosure>(shared_from_base<FuncGraph>(), temp_co… in ToAbstract()
62 AnfNodePtr FuncGraph::output() const { in output()
73 const std::vector<AnfNodePtr> FuncGraph::get_inputs() const { in get_inputs()
86 ParameterPtr FuncGraph::add_parameter() { in add_parameter()
87 FuncGraphPtr this_func_graph = shared_from_base<FuncGraph>(); in add_parameter()
93 void FuncGraph::add_parameter(const ParameterPtr &p) { in add_parameter()
95 manager_.lock()->AddParameter(shared_from_base<FuncGraph>(), p); in add_parameter()
101 ParameterPtr FuncGraph::InsertFrontParameter() { in InsertFrontParameter()
[all …]
|
| D | func_graph_extends.cc | 38 AbstractFunctionPtr FuncGraph::abstract() { in abstract()
58 void FuncGraph::set_output(const AnfNodePtr &value, bool force_new_ret) { in set_output()
62 FuncGraphPtr this_graph = shared_from_base<FuncGraph>(); in set_output()
79 void FuncGraph::DumpFuncGraph(const std::string &path) { in DumpFuncGraph()
81 drawer_(path + ".dot", shared_from_base<FuncGraph>()); in DumpFuncGraph()
85 void FuncGraph::GenerateVarParams(const FuncGraphPtr &specialized_graph, int variable_args_count, in GenerateVarParams()
120 void FuncGraph::GenerateKwParams(const FuncGraphPtr &specialized_graph, in GenerateKwParams()
176 void FuncGraph::GenerateKwargReplNode(const FuncGraphPtr &specialized_graph, in GenerateKwargReplNode()
193 bool FuncGraph::NeedGenerate(const std::vector<abstract::AbstractKeywordArgPtr> &kwarg_list) { in NeedGenerate()
204 void FuncGraph::GenerateDefaultValue(const FuncGraphPtr &specialized_graph, in GenerateDefaultValue()
[all …]
|
| D | func_graph.h | 155 class FuncGraph : public api::FuncGraph, public FuncGraphBase, public EffectInfoHolder {
157 FuncGraph();
160 ~FuncGraph() override = default;
161 MS_DECLARE_PARENT(FuncGraph, FuncGraphBase);
338 std::size_t hash() const override { return std::hash<const FuncGraph *>{}(this); } in hash()
343 if (other.isa<FuncGraph>()) {
|
| /third_party/mindspore/tests/ut/cpp/ir/ |
| D | anf_test.cc | 40 ASSERT_EQ(IsValueNode<FuncGraph>(c), false); in TEST_F()
42 FuncGraphPtr fg = std::make_shared<FuncGraph>(); in TEST_F()
44 ASSERT_EQ(c1.value()->isa<FuncGraph>(), true); in TEST_F()
48 FuncGraphPtr fg = std::make_shared<FuncGraph>(); in TEST_F()
56 FuncGraphPtr fg = std::make_shared<FuncGraph>(); in TEST_F()
70 FuncGraphPtr fg = std::make_shared<FuncGraph>(); in TEST_F()
|
| D | manager_test.cc | 254 FuncGraphPtr func_graph = std::make_shared<FuncGraph>(); in MakeFuncGraph()
277 FuncGraphPtr f = std::make_shared<FuncGraph>(); in MakeNestedGraph()
278 FuncGraphPtr fg = std::make_shared<FuncGraph>(); in MakeNestedGraph()
307 FuncGraphPtr graph_foo = std::make_shared<FuncGraph>(); in MakeNestedGraph2()
314 FuncGraphPtr graph_bar = std::make_shared<FuncGraph>(); in MakeNestedGraph2()
359 FuncGraphPtr func_graph = std::make_shared<FuncGraph>(); in TEST_F()
435 FuncGraphPtr f = std::make_shared<FuncGraph>(); in TEST_F()
436 FuncGraphPtr fg = std::make_shared<FuncGraph>(); in TEST_F()
437 FuncGraphPtr h = std::make_shared<FuncGraph>(); in TEST_F()
497 FuncGraphPtr fg = std::make_shared<FuncGraph>(); in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/pybind_api/ir/ |
| D | func_graph_py.cc | 23 REGISTER_PYBIND_DEFINE(FuncGraph, ([](const pybind11::module *m) { in __anon3b2855100102()
28 (void)py::class_<FuncGraph, FuncGraphPtr>(*m, "FuncGraph") in __anon3b2855100102()
30 .def("str", &FuncGraph::ToString, "Get FuncGraph string representation.") in __anon3b2855100102()
31 … .def("get_return", &FuncGraph::get_return, "Get return node of FuncGraph"); in __anon3b2855100102()
|
| /third_party/mindspore/tests/ut/cpp/utils/ |
| D | graph_utils_test.cc | 35 std::shared_ptr<FuncGraph> GetPythonFunction(std::string function);
45 std::shared_ptr<FuncGraph> g1 = getPyFun("test_graph_utils_isomorphic_1"); in TEST_F()
46 std::shared_ptr<FuncGraph> g2 = getPyFun("test_graph_utils_isomorphic_2"); in TEST_F()
47 std::shared_ptr<FuncGraph> g3 = getPyFun("test_graph_utils_isomorphic_3"); in TEST_F()
|
| /third_party/mindspore/mindspore/core/load_mindir/ |
| D | load_model.h | 26 std::shared_ptr<FuncGraph> LoadMindIR(const std::string &file_name, bool is_lite = false,
29 std::vector<std::shared_ptr<FuncGraph>> LoadMindIRs(const std::vector<std::string> file_names, bool…
34 std::shared_ptr<FuncGraph> ConvertStreamToFuncGraph(const char *buf, const size_t buf_size, bool is…
|
| D | load_model.cc | 202 std::vector<std::shared_ptr<FuncGraph>> LoadMindIRs(std::vector<std::string> file_names, bool is_li… in LoadMindIRs()
205 std::vector<std::shared_ptr<FuncGraph>> funcgraph_vec; in LoadMindIRs()
213 std::shared_ptr<FuncGraph> LoadMindIR(const std::string &file_name, bool is_lite, const unsigned ch… in LoadMindIR()
288 std::shared_ptr<FuncGraph> ConvertStreamToFuncGraph(const char *buf, const size_t buf_size, bool is… in ConvertStreamToFuncGraph()
|
| /third_party/mindspore/tests/ut/cpp/device/ |
| D | hccl_adapter_test.cc | 93 auto graph = std::make_shared<FuncGraph>(); in TEST_F()
116 auto graph = std::make_shared<FuncGraph>(); in TEST_F()
139 auto graph = std::make_shared<FuncGraph>(); in TEST_F()
163 auto graph = std::make_shared<FuncGraph>(); in TEST_F()
187 auto graph = std::make_shared<FuncGraph>(); in TEST_F()
210 auto graph = std::make_shared<FuncGraph>(); in TEST_F()
233 auto graph = std::make_shared<FuncGraph>(); in TEST_F()
257 auto graph = std::make_shared<FuncGraph>(); in TEST_F()
281 auto graph = std::make_shared<FuncGraph>(); in TEST_F()
302 auto graph = std::make_shared<FuncGraph>(); in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/core/api/ir/ |
| D | func_graph.h | 29 class MS_CORE_API FuncGraph {
31 FuncGraph() = default;
32 virtual ~FuncGraph() = default;
|
| D | func_graph_manager.h | 31 class FuncGraph; variable
32 using FuncGraphPtr = std::shared_ptr<FuncGraph>;
|
| /third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/ |
| D | specialize_test.cc | 59 graph_g_ = std::make_shared<FuncGraph>(); in SetUp()
72 graph_f_ = std::make_shared<FuncGraph>(); in SetUp()
91 graph_alpha_ = std::make_shared<FuncGraph>(); in SetUp()
92 graph_beta_ = std::make_shared<FuncGraph>(); in SetUp()
163 FuncGraphPtr graph_g = std::make_shared<FuncGraph>(); in GenerateFromTypes()
192 graph_ = std::make_shared<FuncGraph>(); in SetUp()
|
| D | static_analysis_test.cc | 97 FuncGraphPtr fg = std::make_shared<FuncGraph>(); in GenerateFromTypes()
135 FuncGraphPtr func_graph = std::make_shared<FuncGraph>(); in MakeFuncGraph()
192 graph_g_ = std::make_shared<FuncGraph>(); in SetUp()
205 graph_f_ = std::make_shared<FuncGraph>(); in SetUp()
224 graph_alpha_ = std::make_shared<FuncGraph>(); in SetUp()
225 graph_beta_ = std::make_shared<FuncGraph>(); in SetUp()
332 func_graph_ = std::make_shared<FuncGraph>(); in SetUp()
|
| /third_party/mindspore/mindspore/ccsrc/debug/ |
| D | draw.cc | 134 if (IsValueNode<FuncGraph>(t)) { in DrawEdges()
243 if (IsValueNode<FuncGraph>(node)) { in Shape()
263 if (IsValueNode<FuncGraph>(node)) { in Color()
281 if (node->isa<CNode>() || (node->isa<ValueNode>() && !IsValueNode<FuncGraph>(node))) { in Head()
536 } else if (node->isa<ValueNode>() && !IsValueNode<FuncGraph>(node)) { in Node()
540 if (IsValueNode<FuncGraph>(node)) { in Node()
552 if (IsValueNode<FuncGraph>(node)) { in Node()
605 if (IsValueNode<FuncGraph>(node)) { in Shape()
630 } else if (node->isa<ValueNode>() && !IsValueNode<FuncGraph>(node)) { in Node()
638 if (IsValueNode<FuncGraph>(node)) { in Node()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/debug/rdr/ |
| D | graph_recorder.h | 29 class FuncGraph; variable
30 using FuncGraphPtr = std::shared_ptr<FuncGraph>;
|
| D | running_data_recorder.h | 26 class FuncGraph; variable
28 using FuncGraphPtr = std::shared_ptr<FuncGraph>;
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | specialize_transform.h | 76 if (value->isa<FuncGraph>() || value->isa<Primitive>() || value->isa<parse::NameSpace>()) { in NewReplaceValueNode()
101 if (!IsValueNode<FuncGraph>(inputs[0])) { in operator()
113 if (IsValueNode<FuncGraph>(inputs[i]) || IsValueNode<Primitive>(inputs[i]) || in operator()
|
| D | call_graph_tuple_transform.h | 112 if (IsValueNode<FuncGraph>(node)) { in TransformBranchNode()
123 if (IsValueNode<FuncGraph>(partial_inputs[1])) { in TransformBranchNode()
191 if (!IsValueNode<FuncGraph>(tuple_inputs[i])) { in TransformLayerNode()
|
| D | incorporate_call.h | 132 if (cnode->size() < 1 || !IsValueNode<FuncGraph>(cnode->input(0))) { in Visit()
177 …AnfVisitor::Match(prim::kPrimSwitch, {IsNode, IsValueNode<FuncGraph>, IsValueNode<FuncGraph>})(inp… in operator()
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 72 std::shared_ptr<FuncGraph> anf_graph = MakeFuncGraph(prim, nparam); in MakeDfGraph()
141 FuncGraphPtr anf_graph = std::make_shared<FuncGraph>(); in TEST_F()
327 std::shared_ptr<FuncGraph> anf_graph = MakeFuncGraph(prim, 2); in TEST_F()
338 std::shared_ptr<FuncGraph> anf_graph = MakeFuncGraph(prim, 3); in TEST_F()
349 std::shared_ptr<FuncGraph> anf_graph = MakeFuncGraph(prim, 2); in TEST_F()
360 std::shared_ptr<FuncGraph> anf_graph = MakeFuncGraph(prim, 1); in TEST_F()
371 std::shared_ptr<FuncGraph> anf_graph = MakeFuncGraph(prim, 1); in TEST_F()
382 std::shared_ptr<FuncGraph> anf_graph = MakeFuncGraph(prim, 2); in TEST_F()
395 std::shared_ptr<FuncGraph> anf_graph = MakeFuncGraph(prim, 2); in TEST_F()
407 std::shared_ptr<FuncGraph> anf_graph = MakeFuncGraph(prim, 2); in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | composite.cc | 341 FuncGraphPtr ptr_graph = std::make_shared<FuncGraph>(); in GenerateFromTypes()
420 FuncGraphPtr ret = std::make_shared<FuncGraph>(); in GenerateSequeueFuncGraph()
485 FuncGraphPtr fg = std::make_shared<FuncGraph>(); in GenerateFuncGraph()
498 FuncGraphPtr b = std::make_shared<FuncGraph>(); in GenerateFuncGraph()
526 FuncGraphPtr fg = std::make_shared<FuncGraph>(); in GenerateFuncGraph()
539 FuncGraphPtr b = std::make_shared<FuncGraph>(); in GenerateFuncGraph()
575 FuncGraphPtr k_child = std::make_shared<FuncGraph>(); in GetGrad()
681 grad_fg = std::make_shared<FuncGraph>(); in GenerateFuncGraph()
734 FuncGraphPtr fg_ptr = std::make_shared<FuncGraph>(); in GenerateFuncGraph()
768 FuncGraphPtr fgnext_ptr = std::make_shared<FuncGraph>(); in GenerateFuncGraph()
[all …]
|
| /third_party/mindspore/mindspore/core/base/ |
| D | base.h | 136 class FuncGraph;
137 using FuncGraphPtr = std::shared_ptr<FuncGraph>;
138 using FuncGraphWeakPtr = std::weak_ptr<FuncGraph>;
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 126 if (IsValueNode<FuncGraph>(cnode->input(0))) { in LabelParameterStart()
206 if (!IsValueNode<FuncGraph>(node)) { in Coloring()
249 if (IsValueNode<FuncGraph>(user_node->input(0))) { in BroadCastColoring()
292 if (IsValueNode<FuncGraph>(cnode->input(0))) { in CreateOpInfo()
367 if (IsValueNode<FuncGraph>(temp_cnode->input(0))) { in GetActualOpUsers()
423 if (IsValueNode<FuncGraph>(cnode->input(0))) { in HandleSharedParameter()
467 if (IsValueNode<FuncGraph>(node->input(0))) { in ParameterColoring()
492 if (cnode != nullptr && IsValueNode<FuncGraph>(cnode->input(0))) { in GetShapeType()
520 if (IsValueNode<FuncGraph>(cnode->input(0))) { in FindPipelineCareNode()
674 if (IsValueNode<FuncGraph>(cnode->input(0))) { in InsertReceive()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | segment_runner.cc | 81 if (node->isa<ValueNode>() && !IsValueNode<FuncGraph>(node)) { in RefSubGraphNode()
103 fg = std::make_shared<FuncGraph>(); in TransformSegmentToAnfGraph()
118 !(IsValueNode<FuncGraph>(inps[0]) && in TransformSegmentToAnfGraph()
|