Home
last modified time | relevance | path

Searched defs:graph_proto (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Dproto_exporter.cc324 debugger::GraphProto *graph_proto = model_.mutable_graph(); in GetFuncGraphProtoString() local
335 debugger::GraphProto *graph_proto = model_.mutable_graph(); in GetFuncGraphProto() local
340 …oExporter::ExportFuncGraph(const FuncGraphPtr &func_graph, debugger::GraphProto *const graph_proto, in ExportFuncGraph()
369 …rotoExporter::ExportParameters(const FuncGraphPtr &func_graph, debugger::GraphProto *graph_proto) { in ExportParameters()
390 …rotoExporter::ExportCNodes(const FuncGraphPtr &func_graph, debugger::GraphProto *const graph_proto, in ExportCNodes()
415 … debugger::GraphProto *const graph_proto, LocDebugDumpMode dump_location) { in ExportCNode()
467 debugger::GraphProto *graph_proto) { in ExportFuncGraphOutput()
489 debugger::GraphProto *graph_proto) { in ExportValueNodes()
555 std::string graph_proto = exporter.GetFuncGraphProtoString(func_graph, dump_location); in DumpIRProtoWithSrcInfo() local
Ddebugger.cc542 auto graph_proto = GetGraphProto(graph_ptr); in LoadGraphs() local
566 auto graph_proto = graph_proto_list_.front(); in CheckGraphPtr() local
631 void Debugger::SendGraphAndSuspend(const GraphProto &graph_proto) { in SendGraphAndSuspend()
/third_party/mindspore/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc477 onnx::GraphProto *graph_proto = model_.mutable_graph(); in GetOnnxProtoString() local
490 …nnxExporter::ExportFuncGraph(const FuncGraphPtr &func_graph, onnx::GraphProto *const graph_proto) { in ExportFuncGraph()
511 …nxExporter::ExportParameters(const FuncGraphPtr &func_graph, onnx::GraphProto *const graph_proto) { in ExportParameters()
667 onnx::GraphProto *const graph_proto) { in ExportNodes()
742 … std::map<AnfNodePtr, size_t> *node_map_ptr, onnx::GraphProto *const graph_proto) { in ExportPrimReshape()
775 … std::map<AnfNodePtr, size_t> *node_map_ptr, onnx::GraphProto *const graph_proto) { in ExportPrimReduce()
816 onnx::GraphProto *const graph_proto) { in ExportPrimTranspose()
852 onnx::GraphProto *const graph_proto) { in ExportPrimStridedSlice()
949 onnx::GraphProto *const graph_proto) { in ExportPrimResizeNearestNeighbor()
1010 onnx::GraphProto *const graph_proto) { in ExportPrimExpandDims()
[all …]
Dmindir_exporter.cc181 mind_ir::GraphProto *graph_proto = model_.mutable_graph(); in BuildModel() local
213 …portBuilder::BuildFuncGraph(const FuncGraphPtr &func_graph, mind_ir::GraphProto *const graph_proto, in BuildFuncGraph()
226 …ortBuilder::BuildParameters(const FuncGraphPtr &func_graph, mind_ir::GraphProto *const graph_proto, in BuildParameters()
363 …ExportBuilder::BuildNodes(const FuncGraphPtr &func_graph, mind_ir::GraphProto *const graph_proto) { in BuildNodes()
380 void IrExportBuilder::BuildOutput(const CNodePtr &node, mind_ir::GraphProto *const graph_proto) { in BuildOutput()
477 void IrExportBuilder::BuildCNode(const CNodePtr &node, mind_ir::GraphProto *const graph_proto) { in BuildCNode()
526 …g IrExportBuilder::BuildInputNode(const AnfNodePtr &node, mind_ir::GraphProto *const graph_proto) { in BuildInputNode()
/third_party/mindspore/mindspore/ccsrc/debug/
Ddump_proto.cc387 irpb::GraphProto *graph_proto = model_.mutable_graph(); in GetFuncGraphProtoString() local
392 void ProtoExporter::ExportFuncGraph(const FuncGraphPtr &func_graph, irpb::GraphProto *graph_proto) { in ExportFuncGraph()
410 void ProtoExporter::ExportParameters(const FuncGraphPtr &func_graph, irpb::GraphProto *graph_proto)… in ExportParameters()
429 void ProtoExporter::ExportCNodes(const FuncGraphPtr &func_graph, irpb::GraphProto *graph_proto, in ExportCNodes()
453 … std::map<AnfNodePtr, size_t> *const_map_ptr, irpb::GraphProto *graph_proto) { in ExportCNode()
493 … std::map<AnfNodePtr, size_t> *const_map_ptr, irpb::GraphProto *graph_proto) { in ExportFuncGraphOutput()
519 …r::ExportValueNodes(const std::map<AnfNodePtr, size_t> &const_map, irpb::GraphProto *graph_proto) { in ExportValueNodes()
/third_party/mindspore/mindspore/ccsrc/profiler/device/ascend/
Dmemory_profiling.cc73 GraphMemProto *graph_proto = memory_proto_.add_graph_mem(); in MemoryToPB() local
/third_party/mindspore/mindspore/profiler/parser/
Dcontainer.py127 def __init__(self, graph_proto): argument
Dmemory_usage_parser.py177 def __init__(self, graph_proto, points, framework): argument
/third_party/mindspore/mindspore/core/load_mindir/
Danf_model_parser.cc1102 const auto &graph_proto = model_proto.functions(i); in Parse() local
1119 const auto &graph_proto = model_proto.functions(i); in Parse() local
/third_party/mindspore/mindspore/train/
Dserialization.py38 from mindspore.train.mind_ir_pb2 import GraphProto as graph_proto unknown