| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| 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>…
149 …KernelGraphPtr PreBuildOp(const OpRunInfo &op_run_info, const std::vector<tensor::TensorPtr> &inpu…
163 KernelGraphPtr CreateKernelGraph(const GraphInfo &graph_info, OpRunInfo *op_run_info,
166 static bool DisableLazyBuild(const OpRunInfo &op_run_info);
|
| 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 | session_basic.h | 59 struct OpRunInfo { struct
111 …void RunOp(OpRunInfo *, const GraphInfo &, std::vector<tensor::TensorPtr> *input_tensors, VectorRe…
150 std::shared_ptr<KernelGraph> ConstructSingleOpGraph(const OpRunInfo &op_run_info,
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()
266 …void UpdateOutputAbstract(const std::shared_ptr<KernelGraph> &kernel_graph, OpRunInfo *op_run_info…
275 void GetSingleOpRunInfo(const CNodePtr cnode, OpRunInfo *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 | executor.h | 117 OpRunInfo *op_run_info_{nullptr};
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()
935 bool AscendSession::DisableLazyBuild(const OpRunInfo &op_run_info) { in DisableLazyBuild()
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()
1023 KernelGraphPtr AscendSession::PreBuildOp(const OpRunInfo &op_run_info, in PreBuildOp()
1134 OpRunInfo op_run_info; 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 | 1226 void SessionBasic::GetSingleOpRunInfo(const CNodePtr cnode, OpRunInfo *run_info) { in GetSingleOpRunInfo()
1651 OpRunInfo *op_run_info) const { in UpdateOutputAbstract()
2125 std::shared_ptr<KernelGraph> SessionBasic::ConstructSingleOpGraph(const OpRunInfo &op_run_info, in ConstructSingleOpGraph()
2229 void SessionBasic::RunOp(OpRunInfo *op_run_info, const GraphInfo &graph_info, in RunOp()
2297 OpRunInfo run_info; in RunOpsInGraphImpl()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.h | 38 using OpRunInfo = session::OpRunInfo; variable
111 const ActorInfo &CompileGraph(const OpRunInfo &op_run_info, const GraphInfo &graph_info,
119 …void RunGraph(const ActorInfo &actor_info, OpRunInfo *op_run_info, const std::vector<int64_t> *ten…
|
| D | backend.cc | 185 void UpdateOutputAbstract(const KernelGraphPtr &kernel_graph, OpRunInfo *op_run_info) { in UpdateOutputAbstract()
470 const ActorInfo &MindRTBackend::CompileGraph(const OpRunInfo &op_run_info, const GraphInfo &graph_i… in CompileGraph()
745 OpRunInfo op_run_info; in RunGraphBySingleOp()
1036 void MindRTBackend::RunGraph(const ActorInfo &actor_info, OpRunInfo *op_run_info, in RunGraph()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.h | 40 using session::OpRunInfo;
103 GraphId CompileGraph(const session::OpRunInfo &op_run_info, const GraphInfo &graph_info,
135 OpRunInfo *const run_info, GraphInfo *const graph_info);
|
| D | graph_compiler.cc | 384 GraphId GraphCompiler::CompileGraph(const session::OpRunInfo &op_run_info, const GraphInfo &graph_i… in CompileGraph()
479 … OpRunInfo *const run_info, GraphInfo *const graph_info) { in GetSingleOpRunInfoAndGraphInfo()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | ai_core_dynamic_kernel.cc | 137 optiling::utils::OpRunInfo op_run_info_v2(-1, true, 0); in ComputeTiling()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 1951 session::OpRunInfo op_run_info = {op_exec_info->op_name, in RunOpInMs()
1960 session::OpRunInfo op_run_info = {op_exec_info->op_name, in RunOpInMs()
|