Home
last modified time | relevance | path

Searched refs:func_graph_ (Results 1 – 25 of 50) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dfunction_block.cc39 func_graph_ = std::make_shared<FuncGraph>(); in FunctionBlock()
73 …MS_LOG(DEBUG) << (func_graph_ ? func_graph_->ToString() : "FG(Null)") << " write var `" << var_nam… in WriteVariable()
90 << "/" << (func_graph_ ? func_graph_->ToString() : "FG(Null)") in WriteVariable()
143 MS_LOG(DEBUG) << (func_graph_ ? func_graph_->ToString() : "FG(Null)") << " generate phi node " in ReadVariable()
256 …auto node = func_graph_->NewCNodeInOrder({NewValueNode(prim::kPrimResolve), module_node, symbol_no… in MakeResolve()
265 auto node = func_graph_->NewCNodeInOrder( in MakeInterpret()
276 …MS_LOG(DEBUG) << "graph " << (func_graph_ ? func_graph_->ToString() : "FG(Null)") << " set phi " <… in SetPhiArgument()
281 …MS_LOG(DEBUG) << "remove the phi when call graph " << (func_graph_ ? func_graph_->ToString() : "FG… in SetPhiArgument()
287 …MS_LOG(DEBUG) << "graph " << (func_graph_ ? func_graph_->ToString() : "FG(Null)") << " pred_blocks… in SetPhiArgument()
288 << (pred->func_graph_ ? pred->func_graph_->ToString() : "FG(Null)"); in SetPhiArgument()
[all …]
Dfunction_block.h51 FuncGraphPtr func_graph() { return func_graph_; } in func_graph()
52 std::string ToString() const { return func_graph_->ToString(); } in ToString()
107 FuncGraphPtr func_graph_;
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dstack_frame.h39 func_graph_(func_graph), in StackFrame()
49 …node_slots_ = TopoSort(func_graph_->get_return(), SuccIncoming, [this](const AnfNodePtr &node) -> … in Load()
70 MS_LOG(EXCEPTION) << "The stack frame of " << func_graph_->ToString() in CurrentNode()
80 if (current_node == func_graph_->get_return()) { in NextNode()
89 FuncGraphPtr func_graph() const { return func_graph_; } in func_graph()
97 MS_EXCEPTION_IF_NULL(func_graph_); in ToString()
99 buffer << "StackFrame: " << this << ", " << func_graph_->ToString(); in ToString()
131 FuncGraphPtr func_graph_; variable
Dprogram_specialize.cc123 : specializer_(s), func_graph_(fg), context_(context) { in FuncGraphSpecializer()
214 while (fg != specializer->func_graph_) { in GetTopSpecializer()
221 … << (func_graph_ ? func_graph_->ToString() : "FG(Null)") << ", node: " << node->DebugString() in GetTopSpecializer()
232 << (func_graph_ ? func_graph_->ToString() : "FG(Null)") in GetTopSpecializer()
240 …MS_LOG(DEBUG) << "Before run, origin func graph name: " << (func_graph_ ? func_graph_->ToString() … in Run()
243 …<< (func_graph_ ? func_graph_->get_return() ? func_graph_->get_return()->DebugString() : "return n… in Run()
247 …MS_LOG(DEBUG) << "After run, origin func graph name: " << (func_graph_ ? func_graph_->ToString() :… in Run()
264 if (node->func_graph() != func_graph_) { in FirstPass()
271 MS_LOG(INFO) << "Used top func graph specializer as parent for " << func_graph_->ToString() in FirstPass()
502 …MS_LOG(EXCEPTION) << "Func context is nullptr NodeInfo: " << trace::GetDebugInfo(func_graph_->debu… in BuildSpecializedNodeInner()
[all …]
Dauto_monad.cc1078 : func_graph_(func_graph), manager_(func_graph->manager()), top_(top) {} in AutoMonadConverter()
1084 const auto &info = func_graph_->GetEffectInfo(); in Run()
1093 func_graph_->ClearOrderList(); in Run()
1117 for (auto &cnode : func_graph_->order_list()) { in HandleCNodes()
1120 if (cnode->func_graph() != func_graph_) { in HandleCNodes()
1184 auto make_tuple_node = func_graph_->NewCNode(tuple_inputs); in HandleNoEliminateNodes()
1201 func_graph_->set_output(orig_output); in ClearIsolatedNodes()
1309 auto make_tuple = func_graph_->NewCNode(loads); in InsertLoads()
1333 auto load_cnode = func_graph_->NewCNode({load_prim, ref, u}); in MakeLoad()
1383 auto state_depend = func_graph_->NewCNode({depend, orig_output, node}); in AttachToOutput()
[all …]
Devaluator.cc289 MS_EXCEPTION_IF_NULL(func_graph_); in NormalizeArgs()
290 if (func_graph_->has_flag(FUNC_GRAPH_FLAG_IGNORE_VALUES)) { in NormalizeArgs()
293 MS_LOG(DEBUG) << func_graph_->ToString() << " original: " << mindspore::ToString(args_spec_list) in NormalizeArgs()
301 MS_EXCEPTION_IF_NULL(func_graph_); in BroadenUndeterminedArgs()
302 if (func_graph_->has_flag(FUNC_GRAPH_FLAG_IGNORE_VALUES)) { in BroadenUndeterminedArgs()
306 if (func_graph_->has_flag(kFuncGraphFlagUndetermined)) { in BroadenUndeterminedArgs()
307 func_graph_->set_flag(FUNC_GRAPH_FLAG_IGNORE_VALUES, true); in BroadenUndeterminedArgs()
309 MS_LOG(DEBUG) << "Set " << func_graph_->ToString() << " with IGNORE_VALUES flag."; in BroadenUndeterminedArgs()
Dorder_enforce.cc31 …explicit OrderEnforcer(const FuncGraphPtr &func_graph) : func_graph_(func_graph), manager_(func_gr… in OrderEnforcer()
32 MS_EXCEPTION_IF_NULL(func_graph_); in OrderEnforcer()
52 auto nodes = TopoSort(func_graph_->get_return()); in MakeTopoSortMap()
344 const FuncGraphPtr &func_graph_; member in mindspore::pipeline::__anon18e545450111::OrderEnforcer
/third_party/mindspore/mindspore/core/abstract/
Danalysis_context.cc112 return parent_ == nullptr && func_graph_ == nullptr && args_spec_list_.empty(); in IsDummyContext()
119 if (func_graph_ != other.func_graph_) { in operator ==()
178 AnalysisContextPtr context_new = CreateContext(nullptr, func_graph_, args_broad_shp); in SpecializeKey()
189 if (func_graph_ != nullptr) { in hash()
190 hash_value = hash_combine(hash_value, func_graph_->hash()); in hash()
198 if (func_graph_ != nullptr) { in ToString()
199 buffer << "Func Graph: " << func_graph_->ToString(); in ToString()
217 item->func_graph_ = nullptr; in ClearContext()
Danalysis_context.h41 : parent_(parent), func_graph_(fg), args_spec_list_(args_spec_list) { in AnalysisContext()
57 FuncGraphPtr func_graph() const { return func_graph_; } in func_graph()
68 FuncGraphPtr func_graph_; variable
Dabstract_function.cc161 return func_graph_ == other_fg->func_graph_ && context_ == other_fg->context_ && in operator ==()
166 auto hash_value = hash_combine(tid(), func_graph_->hash()); in hash()
176 MS_EXCEPTION_IF_NULL(func_graph_); in ToString()
179 << "FuncGraph: " << func_graph_->ToString() << "; Context: " << context_->ToString(); in ToString()
/third_party/mindspore/mindspore/ccsrc/debug/rdr/
Dgraph_recorder.h33 GraphRecorder() : BaseRecorder(), func_graph_(nullptr), graph_type_("") {} in GraphRecorder()
36 : BaseRecorder(module, name), func_graph_(graph), graph_type_(file_type) {} in GraphRecorder()
39 void SetFuncGraph(const FuncGraphPtr &func_graph) { func_graph_ = func_graph; } in SetFuncGraph()
45 FuncGraphPtr func_graph_;
Dgraph_recorder.cc70 exporter.ExportFuncGraph(realpath_dat, func_graph_); in Export()
79 DumpIRForRDR(realpath_ir, func_graph_, dump_graph_info_.dump_full_name, dump_mode); in Export()
87 protobuf::DumpIRProto(realpath + ".pb", func_graph_); // save *.pb file in Export()
/third_party/mindspore/mindspore/ccsrc/cxx_api/graph/
Dgraph_data.cc24 : func_graph_(nullptr), om_data_(), model_type_(ModelType::kUnknownType), data_graph_({}) { in GraphData()
28 func_graph_ = func_graph;
33 : func_graph_(nullptr), om_data_(om_data), model_type_(model_type), data_graph_({}) { in GraphData()
62 return func_graph_; in GetFuncGraph()
/third_party/mindspore/mindspore/lite/test/ut/tools/converter/registry/
Dnode_parser_registry_test.cc59 func_graph_ = model_parser->Parse(converter_parameters); in SetUp()
61 api::FuncGraphPtr func_graph_ = nullptr; member in mindspore::NodeParserRegistryTest
65 ASSERT_NE(func_graph_, nullptr); in TEST_F()
66 auto node_list = api::FuncGraph::TopoSort(func_graph_->get_return()); in TEST_F()
Dpass_registry_test.cc45 func_graph_ = model_parser->Parse(converter_parameters); in SetUp()
47 api::FuncGraphPtr func_graph_ = nullptr; member in mindspore::PassRegistryTest
204 ASSERT_NE(func_graph_, nullptr); in TEST_F()
206 auto ret = pass->Execute(func_graph_); in TEST_F()
210 auto node_list = api::FuncGraph::TopoSort(func_graph_->get_return()); in TEST_F()
/third_party/mindspore/mindspore/lite/test/common/
Dimport_from_meta_graphT.cc41 MS_ASSERT(nullptr != func_graph_); in ConverterConstTensor()
48 auto parameter = func_graph_->add_parameter(); in ConverterConstTensor()
113 MS_ASSERT(nullptr != func_graph_); in ConverterCNode()
130 auto new_cnode = func_graph_->NewCNode(op_inputs); in ConverterCNode()
166 return anfImporterFromMetaGraphT.func_graph_; in Fb2Anf()
Dimport_from_meta_graphT.h36 this->func_graph_ = std::make_shared<FuncGraph>(); in AnfImporterFromMetaGraphT()
58 FuncGraphPtr func_graph_; variable
/third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/
Dstatic_analysis_test.cc320 FuncGraphPtr func_graph_; member in mindspore::abstract::TestInferMetaGraph
332 func_graph_ = std::make_shared<FuncGraph>(); in SetUp()
333 ParameterPtr x = func_graph_->add_parameter(); in SetUp()
334 ParameterPtr y = func_graph_->add_parameter(); in SetUp()
340 CNodePtr cnode_add = func_graph_->NewCNode(inputs); in SetUp()
345 CNodePtr cnode_return = func_graph_->NewCNode(inputs); in SetUp()
346 func_graph_->set_return(cnode_return); in SetUp()
357 std::cout << func_graph_->get_return()->ToString() << std::endl; in TEST_F()
362 AbstractBasePtr abs_base_got = engine_->Run(func_graph_, args_spec_list).inferred->abstract(); in TEST_F()
/third_party/mindspore/mindspore/core/ir/
Dfunc_graph.h108 : prim_(prim), func_graph_(FuncGraphWeakPtr(func_graph)) {} in prim_()
111 : prim_(prim), func_graph_(FuncGraphWeakPtr(func_graph)) {} in prim_()
113 FuncGraphTransform(const FuncGraphTransform &t) : prim_(t.prim_), func_graph_(t.func_graph_) {} in FuncGraphTransform()
125 bool IsPrimitive() const { return (func_graph_.lock() == nullptr); } in IsPrimitive()
126 bool IsFuncGraph() const { return (func_graph_.lock() != nullptr); } in IsFuncGraph()
127 FuncGraphPtr func_graph() const { return func_graph_.lock(); } in func_graph()
133 func_graph_ = t.func_graph_;
143 FuncGraphWeakPtr func_graph_; variable
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dresource.h72 FuncGraphPtr func_graph() const { return func_graph_; } in func_graph()
73 void set_func_graph(const FuncGraphPtr &func_graph) { func_graph_ = func_graph; } in set_func_graph()
93 FuncGraphPtr func_graph_; variable
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dmulti_conv_info.cc106 MS_EXCEPTION_IF_NULL(func_graph_); in GetMultiConvNodes()
153 …if (!CreateOutputsOfSplitWithOverlap(func_graph_, conv_nodes_[conv_nodes_.size() - 1], &split_outp… in MultiConvNHSplit()
171 …auto concat_node = CreateOutputsOfConcat(func_graph_, node, split_outputs, &split_info_, conv_cnod… in MultiConvNHSplit()
226 auto conv_cnode = func_graph_->NewCNode(conv_inputs); in CreateNewConvNode()
234 if (!GetMultipleOutputsOfAnfNode(func_graph_, conv_cnode, 1, &tmp_outputs)) { in CreateNewConvNode()
249 func_graph_ = func_graph; in DoSplit()
268 if (!UpdateSplitInfo(func_graph_, conv_nodes_, &split_info_)) { in SplitMultiConv()
Doperator_info.cc54 func_graph_ = func_graph; in Init()
123 …auto tuple_getitem = func_graph_->NewCNode({NewValueNode(std::make_shared<lite::TupleGetItem>()), … in CreateMultipleOutputsOfAnfNode()
152 auto concat_cnode = func_graph_->NewCNode(concat_inputs); in CreateConcateNode()
179 auto addn_cnode = func_graph_->NewCNode(addn_inputs); in CreateReduceNode()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dtrt_converter_context.cc46 std::vector<AnfNodePtr> node_list = TopoSort(func_graph_->get_return()); in Parser()
99 const std::vector<AnfNodePtr> graph_inputs = func_graph_->parameters(); in InitInputTable()
125 MS_EXCEPTION_IF_NULL(func_graph_); in InitValueNodeTable()
126 const std::vector<AnfNodePtr> &node_list = TopoSort(func_graph_->get_return()); in InitValueNodeTable()
223 for (const auto &input_node : func_graph_->parameters()) { in GetGraphInputs()
252 AnfAlgo::GetRealInputs(func_graph_->get_return(), &anf_output_list); in GetGraphOutputs()
Dtrt_converter_context.h43 : func_graph_(fg), in TrtConverterContext()
84 FuncGraphPtr func_graph_; variable
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dgenerate_graph.cc137 func_graph_ = cnode->func_graph(); in Init()
138 if (!func_graph_) { in Init()
142 manager_ = func_graph_->manager(); in Init()
160 CNodePtr cnode = func_graph_->NewCNode(inputs); // using NewCNode to create anfnode in PushBack()

12