Searched refs:subgraph_name (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/profiler/parser/ |
| D | framework_parser.py | 536 subgraph_name = None
546 subgraph_name = self._get_subgraph_name(full_op_name)
547 op_name = self._get_op_name(full_op_name, subgraph_name)
575 return [full_op_name, op_name, op_type, subgraph_name,
588 subgraph_name = full_op_name.split('/', 1)[0]
589 if subgraph_name in ['Default', 'Gradients']:
590 return subgraph_name
593 def _get_op_name(self, full_op_name, subgraph_name): argument
604 if subgraph_name is None:
610 strs = full_op_name.split(subgraph_name + '/')
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_model_parser.cc | 169 const auto subgraph_name = tflite_subgraph->name; in ConvertTfliteGraph() local
176 auto attr_value = MakeValue(subgraph_name); in ConvertTfliteGraph()
205 status = BuildSubFuncGraphMap(idx, func_graph, subgraph_name); in ConvertTfliteGraph()
537 auto subgraph_name = tflite_subgraph->name; in ConvertGraphOutputs() local
539 …output_node->cast<CNodePtr>()->set_fullname_with_scope(subgraph_name + "_output_" + std::to_string… in ConvertGraphOutputs()
542 …output_node->cast<ParameterPtr>()->set_name(subgraph_name + "_output_" + std::to_string(i) + "_par… in ConvertGraphOutputs()
550 const std::string &subgraph_name) { in BuildSubFuncGraphMap() argument
554 if (subgraph_name.find("cond") != std::string::npos) { in BuildSubFuncGraphMap()
556 } else if (subgraph_name.find("body") != std::string::npos) { in BuildSubFuncGraphMap()
560 if (subgraph_name.find("then") != std::string::npos) { in BuildSubFuncGraphMap()
[all …]
|
| D | tflite_model_parser.h | 66 const std::string &subgraph_name);
|
| /third_party/boost/boost/graph/detail/ |
| D | read_graphviz_new.hpp | 50 typedef std::string subgraph_name; typedef
91 std::map< subgraph_name, properties > graph_props; // Root and subgraphs
|
| /third_party/boost/libs/graph/src/ |
| D | read_graphviz_new.cpp | 470 subgraph_name subgraph_ep;
480 static edge_endpoint subgraph(const subgraph_name& ep) in subgraph()
504 static node_or_subgraph_ref subgraphref(const subgraph_name& n) in subgraphref()
526 std::map< subgraph_name, subgraph_info > subgraphs;
769 subgraph_name parse_subgraph(const token& first_token) in parse_subgraph()
799 subgraph_name old_sg = current_subgraph_name; in parse_subgraph()
964 std::set< subgraph_name > done; in get_recursive_members()
974 std::map< subgraph_name, subgraph_info >::const_iterator in get_recursive_members()
1145 std::map< subgraph_name, properties >::const_iterator root_graph_props_i in translate_results_to_graph()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | functionalize_control_op_pass.cc | 28 FuncGraphPtr FunctionalizeControlOpPass::NewFuncGraph(const std::string &subgraph_name, const FmkTy… in NewFuncGraph() argument
34 fg->set_attr("graph_name", MakeValue(subgraph_name)); in NewFuncGraph()
|
| D | functionalize_control_op_pass.h | 37 static FuncGraphPtr NewFuncGraph(const std::string &subgraph_name, const FmkType &fmk_type);
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_model_parser.cc | 734 … const std::string &subgraph_name, const std::string &if_node_name, in ConvertIfSubgraph() argument
744 subgraph->set_attr("graph_name", MakeValue(subgraph_name)); in ConvertIfSubgraph()
754 input_parameter->set_name(subgraph_name + "_input_" + std::to_string(j) + "_parameter"); in ConvertIfSubgraph()
759 …input_parameter->set_name(subgraph_name + "_input_" + std::to_string(j + subgraph_proto.input_size… in ConvertIfSubgraph()
776 return_act_inputs[j]->cast<CNodePtr>()->set_fullname_with_scope(subgraph_name + "_output_" + in ConvertIfSubgraph()
779 return_act_inputs[j]->cast<ParameterPtr>()->set_name(subgraph_name + "_output_" + in ConvertIfSubgraph()
793 std::string subgraph_name; in ConvertIfOnnxNode() local
800 subgraph_name = if_node_name + "_then_branch"; in ConvertIfOnnxNode()
803 …auto status = ConvertIfSubgraph(subgraph_proto, then_branch_graph, subgraph_name, if_node_name, ro… in ConvertIfOnnxNode()
808 subgraph_name = if_node_name + "_else_branch"; in ConvertIfOnnxNode()
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/anf_exporter/ |
| D | anf_exporter.cc | 470 auto subgraph_name = func_graph->get_attr("graph_name"); in ExportSubgraph() local
471 MS_CHECK_TRUE_MSG(subgraph_name != nullptr, RET_ERROR, "subgraph_name is nullptr"); in ExportSubgraph()
473 …_" + std::to_string(meta_graphT->subGraph.size() - 1) + "_" + GetValue<std::string>(subgraph_name); in ExportSubgraph()
|