Home
last modified time | relevance | path

Searched defs:input_parameter (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc205 auto input_parameter = utils::cast<ParameterPtr>(input); in InitSubGraphNode() local
243 auto input_parameter = utils::cast<ParameterPtr>(input); in InitSubGraphInNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dnode_info.cc105 auto input_parameter = input->cast<ParameterPtr>(); in ExtractInputParameterByNode() local
133 auto input_parameter = input->cast<ParameterPtr>(); in ExtractInputParameterNameByNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dpipeline_split.cc108 for (const auto &input_parameter : input_parameters) { in FindForwardGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_model_parser.cc483 auto input_parameter = cond_graph->add_parameter(); in BuildCondGraph() local
929 auto input_parameter = input_anode_iter->second->cast<ParameterPtr>(); in ConvertIfSubgraph() local
934 auto input_parameter = subgraph_extra_inputs[j]->cast<ParameterPtr>(); in ConvertIfSubgraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/
Danf_transform.cc402 auto input_parameter = dyn_cast<Parameter>(graph_input); in RunParallelPass() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/
Dir_grad.cc197 auto input_parameter = ad_param()->fg_->add_parameter(); in IrGrad() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dstep_parallel_utils.cc2537 auto input_parameter = input->cast<ParameterPtr>(); in NodeParameterName() local