| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.h | 103 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);
181 std::unordered_map<GraphInfo, KernelGraphPtr> run_op_graphs_;
|
| D | graph_compiler.cc | 384 GraphId GraphCompiler::CompileGraph(const session::OpRunInfo &op_run_info, const GraphInfo &graph_i… in CompileGraph()
433 KernelGraphPtr GraphCompiler::Fetch(const GraphInfo &graph_info) const { in Fetch()
479 … OpRunInfo *const run_info, GraphInfo *const graph_info) { in GetSingleOpRunInfoAndGraphInfo()
540 void GraphCompiler::EraseSingleOpCache(const GraphInfo &graph_info, const GraphId &graph_id) { in EraseSingleOpCache()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | cpu_session.h | 46 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,
|
| D | gpu_session.h | 48 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,
|
| D | session_basic.h | 50 using GraphInfo = std::string; variable
111 …void RunOp(OpRunInfo *, const GraphInfo &, std::vector<tensor::TensorPtr> *input_tensors, VectorRe…
230 virtual KernelGraphPtr BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info, in BuildOpImpl()
235 virtual void RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImpl()
238 virtual void RunOpImplOrigin(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImplOrigin()
274 …GraphInfo GetSingleOpGraphInfo(const CNodePtr &kernel, const std::vector<tensor::TensorPtr> &input…
319 std::unordered_map<GraphInfo, std::shared_ptr<KernelGraph>> run_op_graphs_;
|
| D | ascend_session.h | 64 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,
|
| D | executor.h | 118 GraphInfo graph_info_;
164 void RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const GraphInfo &graph_info,
|
| D | cpu_session.cc | 223 KernelGraphPtr CPUSession::BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info, in BuildOpImpl()
273 void CPUSession::RunOpImplOrigin(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImplOrigin()
279 void CPUSession::RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImpl()
|
| D | gpu_session.cc | 638 KernelGraphPtr GPUSession::BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info, in BuildOpImpl()
663 void GPUSession::RunOpImplOrigin(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImplOrigin()
669 void GPUSession::RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImpl()
|
| D | ascend_session.cc | 792 KernelGraphPtr AscendSession::BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info, in BuildOpImpl()
914 KernelGraphPtr AscendSession::CreateKernelGraph(const GraphInfo &graph_info, OpRunInfo *op_run_info, in CreateKernelGraph()
942 void AscendSession::RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImpl()
985 void AscendSession::RunOpImplOrigin(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImplOrigin()
1127 std::unordered_map<KernelGraphPtr, GraphInfo> single_op_graphs; in BuildOpsInGraph()
1140 const GraphInfo &graph_info = GetSingleOpGraphInfo(kernel, input_tensor_info.input_tensors); in BuildOpsInGraph()
1153 const GraphInfo &new_graph_info = GetSingleOpGraphInfo(kernel, input_tensor_info.input_tensors); in BuildOpsInGraph()
|
| D | executor.cc | 424 void Executor::RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const GraphInfo &graph_info, in RunOp()
|
| D | session_basic.cc | 1181 GraphInfo SessionBasic::GetSingleOpGraphInfo(const CNodePtr &kernel, in GetSingleOpGraphInfo()
1189 GraphInfo graph_info; in GetSingleOpGraphInfo()
2229 void SessionBasic::RunOp(OpRunInfo *op_run_info, const GraphInfo &graph_info, in RunOp()
2299 GraphInfo graph_info = GetSingleOpGraphInfo(kernel, input_tensor_info.input_tensors); in RunOpsInGraphImpl()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.h | 111 const ActorInfo &CompileGraph(const OpRunInfo &op_run_info, const GraphInfo &graph_info,
156 std::map<GraphInfo, DeviceContext *> graph_info_to_device_context_;
|
| D | backend.cc | 470 const ActorInfo &MindRTBackend::CompileGraph(const OpRunInfo &op_run_info, const GraphInfo &graph_i… in CompileGraph()
746 GraphInfo graph_info; in RunGraphBySingleOp()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.h | 50 struct GraphInfo { struct
55 GraphInfo() = default; argument
56 explicit GraphInfo(std::string id) : cell_id(std::move((id))) {} in GraphInfo() function
58 using GraphInfoPtr = std::shared_ptr<GraphInfo>;
|
| D | pynative_execute.cc | 2190 auto graph_info_cg = std::make_shared<GraphInfo>(cell_id); in InitResourceAndDfBuilder()
2209 auto graph_info_cg = std::make_shared<GraphInfo>(cell_id); in InitResourceAndDfBuilder()
2211 auto graph_info_df = std::make_shared<GraphInfo>(cell_id); in InitResourceAndDfBuilder()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/gnn/ |
| D | graph_data.h | 135 virtual Status GraphInfo(py::dict *out) = 0;
|
| D | graph_data_client.h | 144 Status GraphInfo(py::dict *out) override;
|
| D | graph_data_impl.h | 150 Status GraphInfo(py::dict *out) override;
|
| D | graph_data_client.cc | 290 Status GraphDataClient::GraphInfo(py::dict *out) { in GraphInfo() function in mindspore::dataset::gnn::GraphDataClient
|
| D | graph_data_impl.cc | 612 Status GraphDataImpl::GraphInfo(py::dict *out) { in GraphInfo() function in mindspore::dataset::gnn::GraphDataImpl
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/gnn/ |
| D | bindings.cc | 101 THROW_IF_ERROR(g.GraphInfo(&out)); in __anone4a0376a0102()
|