Home
last modified time | relevance | path

Searched refs:op_run_info (Results 1 – 15 of 15) 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() argument
233 const auto &kernel_graph = ConstructSingleOpGraph(op_run_info, input_tensors, tensors_mask); 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
283 MS_EXCEPTION_IF_NULL(op_run_info); in RunOpImpl()
284 const auto &kernel_graph = BuildOpImpl(*op_run_info, graph_info, *input_tensors, tensors_mask); in RunOpImpl()
306 if (op_run_info->is_dynamic_shape) { in RunOpImpl()
307 UpdateOutputAbstract(kernel_graph, op_run_info); in RunOpImpl()
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>…
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);
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,
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,
Dgpu_session.cc638 KernelGraphPtr GPUSession::BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info, in BuildOpImpl() argument
643 …if (it != run_op_graphs_.end() && kOpCacheBlackList.find(op_run_info.op_name) == kOpCacheBlackList… in BuildOpImpl()
648 const auto &kernel_graph = ConstructSingleOpGraph(op_run_info, input_tensors, tensors_mask); 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
673 MS_EXCEPTION_IF_NULL(op_run_info); in RunOpImpl()
674 const auto &kernel_graph = BuildOpImpl(*op_run_info, graph_info, *input_tensors, tensors_mask); in RunOpImpl()
695 if (op_run_info->is_dynamic_shape) { in RunOpImpl()
696 UpdateOutputAbstract(kernel_graph, op_run_info); in RunOpImpl()
[all …]
Dascend_session.cc792 KernelGraphPtr AscendSession::BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info, in BuildOpImpl() argument
800 const auto &graph = PreBuildOp(op_run_info, input_tensors, tensors_mask); in BuildOpImpl()
914 KernelGraphPtr AscendSession::CreateKernelGraph(const GraphInfo &graph_info, OpRunInfo *op_run_info, in CreateKernelGraph() argument
920 graph = PreBuildOp(*op_run_info, *input_tensors, tensors_mask); in CreateKernelGraph()
926 graph = PreBuildOp(*op_run_info, *input_tensors, tensors_mask); in CreateKernelGraph()
935 bool AscendSession::DisableLazyBuild(const OpRunInfo &op_run_info) { in DisableLazyBuild() argument
938 …return !op_run_info.lazy_build || ms_context->get_param<int>(MS_CTX_EXECUTION_MODE) == kGraphMode … in DisableLazyBuild()
939 … op_run_info.is_dynamic_shape || ms_context->get_param<bool>(MS_CTX_ENABLE_PYNATIVE_SYNCHRONIZE); in DisableLazyBuild()
942 void AscendSession::RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_info, in RunOpImpl() argument
945 MS_EXCEPTION_IF_NULL(op_run_info); in RunOpImpl()
[all …]
Dexecutor.cc424 void Executor::RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const GraphInfo &graph_info, in RunOp() argument
430 MS_EXCEPTION_IF_NULL(op_run_info); in RunOp()
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()
451 task->op_run_info_ = op_run_info; in RunOp()
Dsession_basic.h150 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() 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
266 …dateOutputAbstract(const std::shared_ptr<KernelGraph> &kernel_graph, OpRunInfo *op_run_info) const;
Dsession_basic.cc1651 OpRunInfo *op_run_info) const { in UpdateOutputAbstract()
1653 MS_EXCEPTION_IF_NULL(op_run_info); in UpdateOutputAbstract()
1657 if (AnfAlgo::GetCNodeName(kernel) == op_run_info->op_name) { in UpdateOutputAbstract()
1658 op_run_info->abstract = kernel->abstract(); in UpdateOutputAbstract()
2125 std::shared_ptr<KernelGraph> SessionBasic::ConstructSingleOpGraph(const OpRunInfo &op_run_info, in ConstructSingleOpGraph() argument
2134 PrimitivePtr op_prim = op_run_info.primitive; in ConstructSingleOpGraph()
2159 cnode->set_abstract(op_run_info.abstract); in ConstructSingleOpGraph()
2161 AnfAlgo::SetNodeAttr(kAttrOutputIsDynamicShape, MakeValue(op_run_info.is_dynamic_shape), cnode); in ConstructSingleOpGraph()
2162 if (op_run_info.is_auto_mixed_precision) { in ConstructSingleOpGraph()
2163 AnfAlgo::SetNodeAttr(kAttrPynativeNextOpName, MakeValue(op_run_info.next_op_name), cnode); in ConstructSingleOpGraph()
[all …]
Dexecutor.h164 void RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const GraphInfo &graph_info,
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc185 void UpdateOutputAbstract(const KernelGraphPtr &kernel_graph, OpRunInfo *op_run_info) { in UpdateOutputAbstract() argument
187 MS_EXCEPTION_IF_NULL(op_run_info); in UpdateOutputAbstract()
191 if (AnfAlgo::GetCNodeName(kernel) == op_run_info->op_name) { in UpdateOutputAbstract()
192 op_run_info->abstract = kernel->abstract(); in UpdateOutputAbstract()
470 const ActorInfo &MindRTBackend::CompileGraph(const OpRunInfo &op_run_info, const GraphInfo &graph_i… in CompileGraph() argument
481 …auto graph_id = graph_compiler_->CompileGraph(op_run_info, graph_info, tensors_mask, input_tensors, in CompileGraph()
484 std::string actor_info = std::to_string(graph_id) + "_" + op_run_info.op_name; in CompileGraph()
745 OpRunInfo op_run_info; in RunGraphBySingleOp() local
749 …ph_compiler_->GetSingleOpRunInfoAndGraphInfo(kernel, input_tensor_info.input_tensors, &op_run_info, in RunGraphBySingleOp()
752 …const ActorInfo &actor_info = CompileGraph(op_run_info, graph_info, &input_tensor_info.input_tenso… in RunGraphBySingleOp()
[all …]
Dbackend.h111 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…
/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() argument
499 MS_EXCEPTION_IF_NULL(op_run_info); in ConstructInputTensor()
502 PrimitivePtr op_prim = op_run_info->py_primitive; in ConstructInputTensor()
505 …bool reg_exist = opt::ConstInputToAttrInfoRegistry::Instance().GetRegisterByOpName(op_run_info->op… in ConstructInputTensor()
506 if (op_run_info->is_dynamic_shape && in ConstructInputTensor()
507 …dynamic_shape_const_input_to_attr.find(op_run_info->op_name) == dynamic_shape_const_input_to_attr.… in ConstructInputTensor()
508 MS_LOG(DEBUG) << "current node is dynamic shape: " << op_run_info->op_name; in ConstructInputTensor()
514 if (device_target != kCPUDevice && op_run_info->op_name == prim::kPrimEmbeddingLookup->name()) { in ConstructInputTensor()
518 if (device_target != kGPUDevice && op_run_info->op_name == prim::kPrimGatherD->name()) { in ConstructInputTensor()
524 size_t input_num = op_run_info->op_inputs.size(); in ConstructInputTensor()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.h103 GraphId CompileGraph(const session::OpRunInfo &op_run_info, const GraphInfo &graph_info,
Dgraph_compiler.cc384 GraphId GraphCompiler::CompileGraph(const session::OpRunInfo &op_run_info, const GraphInfo &graph_i… in CompileGraph() argument
399 …KernelGraphPtr graph = session_->ConstructSingleOpGraph(op_run_info, *input_tensors, *tensors_mask… in CompileGraph()