Home
last modified time | relevance | path

Searched refs:DfGraphConvertor (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dconvert.cc110 OpAdapterPtr DfGraphConvertor::FindAdapter(const AnfNodePtr node, bool train) { in FindAdapter()
136 void DfGraphConvertor::InitLoopVar(std::vector<ge::Operator> *init_input) { in InitLoopVar()
210 OpAdapterPtr DfGraphConvertor::FindAdapter(const std::string &name, bool train) { in FindAdapter()
218 void DfGraphConvertor::DrawParamInitSubGraph(const std::string &name, const AnfNodePtr &it) { in DrawParamInitSubGraph()
238 void DfGraphConvertor::SetupParamInitSubGraph(const TensorOrderMap &tensors, std::vector<ge::Operat… in SetupParamInitSubGraph()
288 void DfGraphConvertor::MakeDatasetHandler(const std::string &name, const size_t &input_idx, const A… in MakeDatasetHandler()
304 void DfGraphConvertor::SetupBroadcast(const std::shared_ptr<HcomBroadcast> &broadcast, in SetupBroadcast()
321 void DfGraphConvertor::InitParamWithData(const TensorOrderMap &tensors) { in InitParamWithData()
398 DfGraphConvertor &DfGraphConvertor::InitParam(const TensorOrderMap &tensors) { in InitParam()
416 DfGraphConvertor::MakeDatasetHandler(name, input_idx, it); in InitParam()
[all …]
Dconvert.h53 class DfGraphConvertor {
55 explicit DfGraphConvertor(const AnfGraphPtr &anf_graph) : anf_graph_(anf_graph) { in DfGraphConvertor() function
73 ~DfGraphConvertor() {} in ~DfGraphConvertor()
111 DfGraphConvertor &ConvertAllNode();
112 DfGraphConvertor &BuildGraph();
113 DfGraphConvertor &InitParam(const TensorOrderMap &tensors);
114 DfGraphConvertor &GenerateCheckpointGraph();
115 DfGraphConvertor &GenerateBroadcastGraph(const TensorOrderMap &tensors);
/third_party/mindspore/tests/ut/cpp/transform/
Dconvert_test.cc62 DfGraphConvertor converter(func_graph); in TEST_F()
75 DfGraphConvertor converter(anf_graph); in MakeDfGraph()
98 DfGraphConvertor converter(anf_graph); in TEST_F()
109 DfGraphConvertor converter(anf_graph); in TEST_F()
131 DfGraphConvertor converter(anfGraph); in TEST_F()
167 DfGraphConvertor converter(anf_graph); in TEST_F()
195 DfGraphConvertor converter(anf_graph); in TEST_F()
223 DfGraphConvertor converter(anf_graph); in TEST_F()
246 DfGraphConvertor converter(anf_graph); in TEST_F()
268 DfGraphConvertor converter(anf_graph); in TEST_F()
[all …]
Dgraph_runner_test.cc57 std::shared_ptr<DfGraphConvertor> MakeGeGraph() { in MakeGeGraph()
73 return std::make_shared<DfGraphConvertor>(anf_graph); in MakeGeGraph()
145 std::shared_ptr<DfGraphConvertor> converter = MakeGeGraph(); in TEST_F()
179 std::shared_ptr<DfGraphConvertor> converter = MakeGeGraph(); in TEST_F()
217 std::shared_ptr<DfGraphConvertor> converter = MakeGeGraph(); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute_ge.cc83 transform::OpAdapterPtr adapter = transform::DfGraphConvertor::FindAdapter(op_name, true); in SetInputsForSingleOpGraph()
126 transform::OpAdapterPtr adapter = transform::DfGraphConvertor::FindAdapter(op_name, true); in BuildSingleOpGraph()
272 transform::OpAdapterPtr adapter = transform::DfGraphConvertor::FindAdapter(op_name, true); in RunOpInGE()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/
Dmodel_converter.cc76 transform::DfGraphConvertor converter(anf_graph); in ConvertFuncGraphToAIR()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline_ge.cc43 using mindspore::transform::DfGraphConvertor;
180 DfGraphConvertor converter(anf_graph); in AddDFGraph()
/third_party/mindspore/mindspore/ccsrc/vm/
Dgraph_partition.cc658 auto adpt = transform::DfGraphConvertor::FindAdapter(name); in IsCut()