Home
last modified time | relevance | path

Searched refs:graph_info (Results 1 – 21 of 21) sorted by relevance

/third_party/mindspore/tests/ut/python/dataset/
Dtest_graphdata.py144 graph_info = g.graph_info()
145 assert graph_info['node_type'] == [1, 2]
146 assert graph_info['edge_type'] == [0]
147 assert graph_info['node_num'] == {1: 10, 2: 10}
148 assert graph_info['edge_num'] == {0: 40}
149 assert graph_info['node_feature_type'] == [1, 2, 3, 4]
150 assert graph_info['edge_feature_type'] == [1, 2]
179 return self.g.graph_info()['edge_num'][0] // self.batch_num
211 edge_num = g.graph_info()['edge_num'][0]
Dtest_graphdata_distributed.py63 return self.g.graph_info()['edge_num'][0] // self.batch_num
129 edge_num = g.graph_info()['edge_num'][0]
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc384 …hId GraphCompiler::CompileGraph(const session::OpRunInfo &op_run_info, const GraphInfo &graph_info, in CompileGraph() argument
389 auto iter = run_op_graphs_.find(graph_info); in CompileGraph()
414 run_op_graphs_[graph_info] = graph; in CompileGraph()
433 KernelGraphPtr GraphCompiler::Fetch(const GraphInfo &graph_info) const { in Fetch()
434 auto iter = run_op_graphs_.find(graph_info); in Fetch()
436 MS_LOG(ERROR) << "Can't find graph for: " << graph_info; in Fetch()
479 … OpRunInfo *const run_info, GraphInfo *const graph_info) { in GetSingleOpRunInfoAndGraphInfo() argument
481 MS_EXCEPTION_IF_NULL(graph_info); in GetSingleOpRunInfoAndGraphInfo()
483 *graph_info = session_->GetSingleOpGraphInfo(kernel, input_tensors); in GetSingleOpRunInfoAndGraphInfo()
540 void GraphCompiler::EraseSingleOpCache(const GraphInfo &graph_info, const GraphId &graph_id) { in EraseSingleOpCache() argument
[all …]
Dgraph_compiler.h103 GraphId CompileGraph(const session::OpRunInfo &op_run_info, const GraphInfo &graph_info,
111 KernelGraphPtr Fetch(const GraphInfo &graph_info) const;
135 OpRunInfo *const run_info, GraphInfo *const graph_info);
168 void EraseSingleOpCache(const GraphInfo &graph_info, const GraphId &graph_id);
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dcpu_session.h46 KernelGraphPtr BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info,
49 …void RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_info, std::vector<tensor::TensorPtr>…
51 void RunOpImplOrigin(const GraphInfo &graph_info, OpRunInfo *op_run_info,
Dcpu_session.cc223 KernelGraphPtr CPUSession::BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info, in BuildOpImpl() argument
227 auto it = run_op_graphs_.find(graph_info); in BuildOpImpl()
241 run_op_graphs_[graph_info] = kernel_graph; in BuildOpImpl()
273 void CPUSession::RunOpImplOrigin(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImplOrigin() argument
276 RunOpImpl(graph_info, op_run_info, input_tensors, outputs, tensors_mask); in RunOpImplOrigin()
279 void CPUSession::RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImpl() argument
284 const auto &kernel_graph = BuildOpImpl(*op_run_info, graph_info, *input_tensors, tensors_mask); in RunOpImpl()
Dgpu_session.h48 KernelGraphPtr BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info,
51 …void RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_info, std::vector<tensor::TensorPtr>…
53 void RunOpImplOrigin(const GraphInfo &graph_info, OpRunInfo *op_run_info,
Dascend_session.h64 KernelGraphPtr BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info,
69 void RunOpImplOrigin(const GraphInfo &graph_info, OpRunInfo *op_run_info,
73 …void RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_info, std::vector<tensor::TensorPtr>…
163 KernelGraphPtr CreateKernelGraph(const GraphInfo &graph_info, OpRunInfo *op_run_info,
Dgpu_session.cc638 KernelGraphPtr GPUSession::BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info, in BuildOpImpl() argument
642 auto it = run_op_graphs_.find(graph_info); in BuildOpImpl()
658 run_op_graphs_[graph_info] = kernel_graph; in BuildOpImpl()
663 void GPUSession::RunOpImplOrigin(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImplOrigin() argument
666 RunOpImpl(graph_info, op_run_info, input_tensors, outputs, tensors_mask); in RunOpImplOrigin()
669 void GPUSession::RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImpl() argument
674 const auto &kernel_graph = BuildOpImpl(*op_run_info, graph_info, *input_tensors, tensors_mask); in RunOpImpl()
700 run_op_graphs_.erase(graph_info); in RunOpImpl()
Dpynative_task_manager.h33 …RunOpContext(std::string graph_info, bool is_dynamic_shape, KernelGraphPtr graph, std::vector<int6… in RunOpContext() argument
36 : graph_info_(std::move(graph_info)), in RunOpContext()
Dexecutor.cc424 void Executor::RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const GraphInfo &graph_info, in RunOp() argument
443 session->RunOpImpl(graph_info, op_run_info, input_tensors, outputs, tensors_mask); in RunOp()
445 session->RunOpImpl(graph_info, op_run_info, input_tensors, outputs, tensors_mask); in RunOp()
452 task->graph_info_ = graph_info; in RunOp()
Dascend_session.cc792 KernelGraphPtr AscendSession::BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info, in BuildOpImpl() argument
795 auto it = run_op_graphs_.find(graph_info); in BuildOpImpl()
809 run_op_graphs_[graph_info] = graph; in BuildOpImpl()
914 KernelGraphPtr AscendSession::CreateKernelGraph(const GraphInfo &graph_info, OpRunInfo *op_run_info, in CreateKernelGraph() argument
923 run_op_graphs_[graph_info] = graph; in CreateKernelGraph()
929 graph = run_op_graphs_[graph_info]; in CreateKernelGraph()
942 void AscendSession::RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImpl() argument
948 RunOpImplOrigin(graph_info, op_run_info, input_tensors, outputs, tensors_mask); in RunOpImpl()
953 bool cache_miss = run_op_graphs_.find(graph_info) == run_op_graphs_.end(); in RunOpImpl()
954 auto graph = CreateKernelGraph(graph_info, op_run_info, input_tensors, tensors_mask, cache_miss); in RunOpImpl()
[all …]
Dsession_basic.h230 virtual KernelGraphPtr BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info, in BuildOpImpl() argument
235 virtual void RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImpl() argument
238 virtual void RunOpImplOrigin(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImplOrigin() argument
Dsession_basic.cc1189 GraphInfo graph_info; in GetSingleOpGraphInfo() local
1195 … [&](const auto &dim) { (void)graph_info.append(std::to_string(dim) + "_"); }); in GetSingleOpGraphInfo()
1196 (void)graph_info.append(std::to_string(tensor->data_type()) + "_"); in GetSingleOpGraphInfo()
1199 (void)graph_info.append(std::to_string(type_id) + "_"); in GetSingleOpGraphInfo()
1201 (void)graph_info.append(format + "_"); in GetSingleOpGraphInfo()
1204 (void)graph_info.append(std::to_string(padding_type) + "_"); in GetSingleOpGraphInfo()
1213 (void)graph_info.append(element.second->ToString() + "_"); in GetSingleOpGraphInfo()
1217 (void)graph_info.append(build_shape->ToString() + "_"); in GetSingleOpGraphInfo()
1220 (void)graph_info.append(std::to_string(output_type) + "_"); in GetSingleOpGraphInfo()
1222 graph_info.append(std::to_string(prim->id())); in GetSingleOpGraphInfo()
[all …]
Dexecutor.h164 void RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const GraphInfo &graph_info,
/third_party/mindspore/mindspore/profiler/parser/
Dframework_parser.py482 for graph_info in graph_file:
487 if graph_info.startswith("op_name:") is False:
488 pre_graph_info = pre_graph_info + graph_info
492 pre_graph_info = graph_info
504 def _parse_graph_row_and_save(self, task_cache, csv_writer, graph_info): argument
513 result = self._parse_one_row_graph_info(graph_info)
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute.h261 auto &graph_info = top_cell()->graph_info_map()[g]; in SetParamNodeMapInGraphInfoMap() local
262 MS_EXCEPTION_IF_NULL(graph_info); in SetParamNodeMapInGraphInfoMap()
263 graph_info->params[id] = param; in SetParamNodeMapInGraphInfoMap()
267 auto &graph_info = top_cell()->graph_info_map()[g]; variable
268 MS_EXCEPTION_IF_NULL(graph_info);
269 graph_info->node_map[id] = std::make_pair(node, std::vector<int64_t>{index});
273 auto &graph_info = top_cell()->graph_info_map()[g]; in SetNodeMapInGraphInfoMap() local
274 MS_EXCEPTION_IF_NULL(graph_info); in SetNodeMapInGraphInfoMap()
275 graph_info->node_map[id] = std::make_pair(node, index); in SetNodeMapInGraphInfoMap()
Dpynative_execute.cc312 auto &graph_info = *graph_info_key; in GetSingleOpGraphInfo() local
368 graph_info = buf.str(); in GetSingleOpGraphInfo()
1366 auto graph_info = top_cell()->graph_info_map().at(df_builder); in GetInput() local
1367 MS_EXCEPTION_IF_NULL(graph_info); in GetInput()
1368 if (graph_info->params.find(obj_id) == graph_info->params.end()) { in GetInput()
1381 node = graph_info->params.at(obj_id); in GetInput()
1421 auto graph_info = top_cell()->graph_info_map().at(curr_g_); in GetObjNode() local
1422 MS_EXCEPTION_IF_NULL(graph_info); in GetObjNode()
1423 const auto &out = graph_info->node_map.at(obj_id); in GetObjNode()
1430 if (graph_info->params.find(obj_id) != graph_info->params.end()) { in GetObjNode()
[all …]
/third_party/mindspore/mindspore/dataset/engine/
Dgraphdata.py454 def graph_info(self): member in GraphData
465 return self._graph_data.graph_info()
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc470 …t ActorInfo &MindRTBackend::CompileGraph(const OpRunInfo &op_run_info, const GraphInfo &graph_info, in CompileGraph() argument
481 …auto graph_id = graph_compiler_->CompileGraph(op_run_info, graph_info, tensors_mask, input_tensors, in CompileGraph()
494 graph_info_to_device_context_[graph_info] = device_context; in CompileGraph()
746 GraphInfo graph_info; in RunGraphBySingleOp() local
750 &graph_info); in RunGraphBySingleOp()
752 …const ActorInfo &actor_info = CompileGraph(op_run_info, graph_info, &input_tensor_info.input_tenso… in RunGraphBySingleOp()
1030 const auto &graph_info = graph_info_to_device_context_.begin()->first; in EraseSingleOpCache() local
1032 graph_compiler_->EraseSingleOpCache(graph_info, graph->graph_id()); in EraseSingleOpCache()
Dbackend.h111 const ActorInfo &CompileGraph(const OpRunInfo &op_run_info, const GraphInfo &graph_info,