Home
last modified time | relevance | path

Searched defs:op_run_info (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/session/
Dcpu_session.cc223 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()
Dsession_basic.h230 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()
Dgpu_session.cc638 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()
Dascend_session.cc792 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
Dexecutor.cc424 void Executor::RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const GraphInfo &graph_info, in RunOp()
Dsession_basic.cc2125 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/
Dbackend.cc185 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/
Dgraph_compiler.cc384 GraphId GraphCompiler::CompileGraph(const session::OpRunInfo &op_run_info, const GraphInfo &graph_i… in CompileGraph()
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute.cc497 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