Home
last modified time | relevance | path

Searched refs:parameter_nodes (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/utils/
Dcallbacks_ge.cc46 auto parameter_nodes = graph->parameters(); in GetParameterShape() local
47 for (auto &node : parameter_nodes) { in GetParameterShape()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dgraph_partitioner.cc270 std::vector<AnfNodePtr> parameter_nodes = fg->parameters(); in CreateNewGraph() local
271 for (size_t i = 0; i < parameter_nodes.size(); i++) { in CreateNewGraph()
272 ParameterPtr dst = parameter_nodes[i]->cast<ParameterPtr>(); in CreateNewGraph()