Searched defs:op_run_info (Results 1 – 9 of 9) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/backend/session/ |
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 | session_basic.h | 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()
|
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() local
|
D | executor.cc | 424 void Executor::RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const GraphInfo &graph_info, in RunOp()
|
D | session_basic.cc | 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()
|
/third_party/mindspore/mindspore/ccsrc/vm/ |
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() local 1036 void MindRTBackend::RunGraph(const ActorInfo &actor_info, OpRunInfo *op_run_info, in RunGraph()
|
/third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
D | graph_compiler.cc | 384 GraphId GraphCompiler::CompileGraph(const session::OpRunInfo &op_run_info, const GraphInfo &graph_i… in CompileGraph()
|
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
D | pynative_execute.cc | 497 void ConstructInputTensor(const OpExecInfoPtr &op_run_info, std::vector<int64_t> *tensors_mask, in ConstructInputTensor() 546 void ConvertAttrToUnifyMindIR(const OpExecInfoPtr &op_run_info) { in ConvertAttrToUnifyMindIR() 1951 session::OpRunInfo op_run_info = {op_exec_info->op_name, in RunOpInMs() local 1960 session::OpRunInfo op_run_info = {op_exec_info->op_name, in RunOpInMs() local
|