Home
last modified time | relevance | path

Searched defs:anf_graph (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/tests/ut/cpp/transform/
Dconvert_test.cc72 std::shared_ptr<FuncGraph> anf_graph = MakeFuncGraph(prim, nparam); in MakeDfGraph() local
96 FuncGraphPtr anf_graph = MakeFuncGraph(conv2d, 2); in TEST_F() local
106 FuncGraphPtr anf_graph = MakeFuncGraph(prim, 5); // ary, ksize, stride, padding, data_format in TEST_F() local
141 FuncGraphPtr anf_graph = std::make_shared<FuncGraph>(); in TEST_F() local
194 auto anf_graph = *(manager->func_graphs().begin()); in TEST_F() local
222 auto anf_graph = *(manager->func_graphs().begin()); in TEST_F() local
245 auto anf_graph = *(manager->func_graphs().begin()); in TEST_F() local
267 auto anf_graph = *(manager->func_graphs().begin()); in TEST_F() local
289 auto anf_graph = *(manager->func_graphs().begin()); in TEST_F() local
317 auto anf_graph = *(manager->func_graphs().begin()); in TEST_F() local
[all …]
Dgraph_runner_test.cc70 FuncGraphPtr anf_graph = MakeFuncGraph(conv2d, 2); in MakeGeGraph() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_model_parser.cc68 ParameterPtr CreateConstParamter(const FuncGraphPtr &anf_graph, int val) { in CreateConstParamter()
112 STATUS AddIterNumsUpdateEdge(const FuncGraphPtr &anf_graph, std::vector<AnfNodePtr> *return_new_inp… in AddIterNumsUpdateEdge()
162 STATUS BuildReturnNode(const FuncGraphPtr &anf_graph, const std::vector<AnfNodePtr> &return_inputs)… in BuildReturnNode()
209 STATUS BuildOpOutputs(const onnx::NodeProto &onnx_node, const FuncGraphPtr &anf_graph, in BuildOpOutputs()
318 STATUS ConvertGraphOutputs(const onnx::GraphProto &onnx_graph, const FuncGraphPtr &anf_graph, in ConvertGraphOutputs()
619 …nnxModelParser::ConvertOnnxGraph(const onnx::GraphProto &onnx_graph, const FuncGraphPtr &anf_graph, in ConvertOnnxGraph()
668 …US OnnxModelParser::ConvertNodes(const onnx::GraphProto &onnx_graph, const FuncGraphPtr &anf_graph, in ConvertNodes()
844 STATUS OnnxModelParser::BuildCNode(const onnx::NodeProto &onnx_node, const FuncGraphPtr &anf_graph, in BuildCNode()
1139 STATUS OnnxModelParser::AddTensorArrayEdge(const FuncGraphPtr &anf_graph, std::vector<AnfNodePtr> *… in AddTensorArrayEdge()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/
Dmodel_converter.cc28 transform::TensorOrderMap GetParams(const FuncGraphPtr &anf_graph) { in GetParams()
74 transform::DfGraphPtr ModelConverter::ConvertFuncGraphToAIR(const FuncGraphPtr &anf_graph) { in ConvertFuncGraphToAIR()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconvert_tuple_output_to_maketuple.cc37 FuncGraphPtr anf_graph = tuple_anf->func_graph(); in ConvertTupleInputToMakeTuple() local
/third_party/mindspore/tests/ut/cpp/session/
Dsession_basic_test.cc42 auto anf_graph = std::make_shared<FuncGraph>(); in TEST_F() local
Dkernel_graph_test.cc60 auto anf_graph = std::make_shared<FuncGraph>(); in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline_ge.cc179 FuncGraphPtr anf_graph = info.at(phase)->func_graph; in AddDFGraph() local
246 FuncGraphPtr anf_graph = info.at(phase)->func_graph; in BuildDFGraph() local
489 FuncGraphPtr anf_graph = info.at(phase)->func_graph; in ExecDFGraph() local
/third_party/mindspore/mindspore/ccsrc/cxx_api/
Dserialization.cc113 FuncGraphPtr anf_graph = nullptr; in Load() local
176 FuncGraphPtr anf_graph; in Load() local
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dconvert.h55 explicit DfGraphConvertor(const AnfGraphPtr &anf_graph) : anf_graph_(anf_graph) { in DfGraphConvertor()
Dconvert.cc1376 FuncGraphPtr anf_graph = graph_node->value()->cast<FuncGraphPtr>(); in ProcessSubgraph() local
/third_party/mindspore/mindspore/ccsrc/utils/
Dcallbacks_ge.cc72 FuncGraphPtr anf_graph = transform::DfGraphManager::GetInstance().GetAnfGraph(graph_id); in GetMeTensorTransformed() local
/third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/
Dmem_reuse_test.cc153 auto anf_graph = std::make_shared<FuncGraph>(); in CreateGraphWithExecOrder() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc475 const FuncGraphPtr &anf_graph, in ConvertGraphInputsAndConsts()
856 const FuncGraphPtr &anf_graph, int output_size) { in ConvertOutputTensor()
1140 …::MakeAnfGraphOutputs(const std::vector<AnfNodePtr> &output_nodes, const FuncGraphPtr &anf_graph) { in MakeAnfGraphOutputs()