Home
last modified time | relevance | path

Searched refs:graph_proto (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc374 void ExportFuncGraph(const FuncGraphPtr &func_graph, onnx::GraphProto *graph_proto);
375 void ExportParameters(const FuncGraphPtr &func_graph, onnx::GraphProto *graph_proto);
379 onnx::GraphProto *graph_proto);
388 onnx::GraphProto *graph_proto);
391 onnx::GraphProto *graph_proto);
394 std::map<AnfNodePtr, size_t> *node_map_ptr, onnx::GraphProto *graph_proto);
396 std::map<AnfNodePtr, size_t> *node_map_ptr, onnx::GraphProto *graph_proto);
398 … std::map<AnfNodePtr, size_t> *node_map_ptr, onnx::GraphProto *graph_proto);
400 … std::map<AnfNodePtr, size_t> *node_map_ptr, onnx::GraphProto *graph_proto);
402 … std::map<AnfNodePtr, size_t> *node_map_ptr, onnx::GraphProto *graph_proto);
[all …]
Dmindir_exporter.cc97 void BuildFuncGraph(const FuncGraphPtr &func_graph, mind_ir::GraphProto *const graph_proto,
99 void BuildParameters(const FuncGraphPtr &func_graph, mind_ir::GraphProto *const graph_proto,
101 void BuildNodes(const FuncGraphPtr &func_graph, mind_ir::GraphProto *const graph_proto);
102 void BuildOutput(const CNodePtr &node, mind_ir::GraphProto *const graph_proto);
103 void BuildCNode(const CNodePtr &node, mind_ir::GraphProto *const graph_proto);
104 std::string BuildInputNode(const AnfNodePtr &node, mind_ir::GraphProto *const graph_proto);
181 mind_ir::GraphProto *graph_proto = model_.mutable_graph(); in BuildModel() local
182 graph_proto->set_name(func_graph->ToString()); in BuildModel()
183 graph_proto->set_bprop_hash(func_graph->bprop_hash()); in BuildModel()
190 BuildFuncGraph(func_graph, graph_proto, save_tensor_data); in BuildModel()
[all …]
/third_party/mindspore/mindspore/ccsrc/debug/
Ddump_proto.cc54 void ExportFuncGraph(const FuncGraphPtr &func_graph, irpb::GraphProto *graph_proto);
55 void ExportParameters(const FuncGraphPtr &func_graph, irpb::GraphProto *graph_proto);
56 void ExportCNodes(const FuncGraphPtr &func_graph, irpb::GraphProto *graph_proto,
59 std::map<AnfNodePtr, size_t> *const_map_ptr, irpb::GraphProto *graph_proto);
62 irpb::GraphProto *graph_proto);
63 …oid ExportValueNodes(const std::map<AnfNodePtr, size_t> &const_map, irpb::GraphProto *graph_proto);
387 irpb::GraphProto *graph_proto = model_.mutable_graph(); in GetFuncGraphProtoString() local
388 ExportFuncGraph(func_graph, graph_proto); in GetFuncGraphProtoString()
392 void ProtoExporter::ExportFuncGraph(const FuncGraphPtr &func_graph, irpb::GraphProto *graph_proto) { in ExportFuncGraph() argument
393 if (func_graph == nullptr || graph_proto == nullptr) { in ExportFuncGraph()
[all …]
/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Dproto_exporter.cc324 debugger::GraphProto *graph_proto = model_.mutable_graph(); in GetFuncGraphProtoString() local
325 ExportFuncGraph(func_graph, graph_proto, dump_location); in GetFuncGraphProtoString()
335 debugger::GraphProto *graph_proto = model_.mutable_graph(); in GetFuncGraphProto() local
336 ExportFuncGraph(func_graph, graph_proto); in GetFuncGraphProto()
340 …oExporter::ExportFuncGraph(const FuncGraphPtr &func_graph, debugger::GraphProto *const graph_proto, in ExportFuncGraph() argument
342 if (func_graph == nullptr || graph_proto == nullptr) { in ExportFuncGraph()
350 graph_proto->set_name(func_graph->ToString()); in ExportFuncGraph()
360 graph_proto->set_root_name(std::to_string(root_graph_id)); in ExportFuncGraph()
362 ExportParameters(func_graph, graph_proto); in ExportFuncGraph()
364 ExportCNodes(func_graph, graph_proto, &const_map, dump_location); in ExportFuncGraph()
[all …]
Dproto_exporter.h52 void ExportFuncGraph(const FuncGraphPtr &func_graph, debugger::GraphProto *const graph_proto,
54 void ExportParameters(const FuncGraphPtr &func_graph, debugger::GraphProto *graph_proto);
55 void ExportCNodes(const FuncGraphPtr &func_graph, debugger::GraphProto *const graph_proto,
58 … std::map<AnfNodePtr, size_t> *const_map_ptr, debugger::GraphProto *const graph_proto,
62 debugger::GraphProto *graph_proto);
63 …ExportValueNodes(const std::map<AnfNodePtr, size_t> &const_map, debugger::GraphProto *graph_proto);
Ddebugger.cc542 auto graph_proto = GetGraphProto(graph_ptr); in LoadGraphs() local
544 graph_proto_list_.push_back(graph_proto); in LoadGraphs()
566 auto graph_proto = graph_proto_list_.front(); in CheckGraphPtr() local
567 SendGraphAndSuspend(graph_proto); in CheckGraphPtr()
631 void Debugger::SendGraphAndSuspend(const GraphProto &graph_proto) { in SendGraphAndSuspend() argument
635 EventReply reply = grpc_client_->SendGraph(graph_proto); in SendGraphAndSuspend()
Ddebugger.h188 void SendGraphAndSuspend(const GraphProto &graph_proto);
/third_party/mindspore/mindspore/ccsrc/profiler/device/ascend/
Dmemory_profiling.cc73 GraphMemProto *graph_proto = memory_proto_.add_graph_mem(); in MemoryToPB() local
74 if (graph_proto == nullptr) { in MemoryToPB()
78 graph_proto->set_graph_id(graph.second->GetGraphId()); in MemoryToPB()
79 graph_proto->set_static_mem(graph.second->GetStaticMemSize()); in MemoryToPB()
82 NodeMemProto *node_mem = graph_proto->add_node_mems(); in MemoryToPB()
101 TensorMemProto *tensor_mem = graph_proto->add_tensor_mems(); in MemoryToPB()
/third_party/mindspore/mindspore/profiler/parser/
Dcontainer.py127 def __init__(self, graph_proto): argument
128 self._graph_proto = graph_proto
129 self.graph_id = graph_proto.graph_id
130 self.static_mem = graph_proto.static_mem / GIGABYTES
Dmemory_usage_parser.py115 for graph_proto in graphs:
116 graph_id = graph_proto.graph_id
121 graph_parser = GraphMemoryParser(graph_proto, self._points, self._framework)
177 def __init__(self, graph_proto, points, framework): argument
183 self._graph_proto = graph_proto
/third_party/mindspore/mindspore/train/summary/
Dsummary_record.py331 graph_proto = self.network.get_func_graph_proto()
332 if graph_proto is None and train_network is not None:
333 graph_proto = train_network.get_func_graph_proto()
334 if graph_proto is None:
337 self._event_writer.write({'graph': [{'step': step, 'value': graph_proto}]})
/third_party/mindspore/mindspore/core/load_mindir/
Danf_model_parser.cc1102 const auto &graph_proto = model_proto.functions(i); in Parse() local
1103 if (!graph_proto.has_name()) { in Parse()
1106 if (anfnode_build_map_.count(graph_proto.name()) > 0) { in Parse()
1107 MS_LOG(EXCEPTION) << "There is a duplication function graph name: " << graph_proto.name(); in Parse()
1109 anfnode_build_map_[graph_proto.name()] = std::make_shared<ValueNode>(graph); in Parse()
1119 const auto &graph_proto = model_proto.functions(i); in Parse() local
1120 FuncGraphPtr graph = GetValueNode<FuncGraphPtr>(anfnode_build_map_[graph_proto.name()]); in Parse()
1121 if (!BuildFuncGraph(graph, graph_proto)) { in Parse()
1125 MS_LOG(DEBUG) << "Parse pb to build FuncGraph Success! " << graph_proto.name(); in Parse()
/third_party/mindspore/mindspore/train/callback/
D_summary_collector.py537 graph_proto = network.get_func_graph_proto()
538 if graph_proto is None:
543 self._record.add_value(PluginEnum.GRAPH.value, 'train_network/auto', graph_proto)
/third_party/mindspore/mindspore/train/
Dserialization.py38 from mindspore.train.mind_ir_pb2 import GraphProto as graph_proto unknown
844 graphproto = graph_proto()