| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/ |
| D | common_backend_optimization.cc | 77 void BackendCommonOptimization(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in BackendCommonOptimization()
104 void OpBackendCommonOptimization(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in OpBackendCommonOptimization()
118 void CommonFinalOptimization(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in CommonFinalOptimization()
148 void CommonUnifyMindIR(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in CommonUnifyMindIR()
172 void AddDynamicShapeAttrPass(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in AddDynamicShapeAttrPass()
187 void EliminateIllegalDataTypePass(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in EliminateIllegalDataTypePass()
212 void DynamicShapeConvertPass(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in DynamicShapeConvertPass()
239 void OptimizationWithoutBackend(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in OptimizationWithoutBackend()
264 void OptimizationForAnyTypeKernelGraph(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in OptimizationForAnyTypeKernelGraph()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/session/ |
| D | anf_runtime_algorithm_test.cc | 49 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
108 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
119 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
131 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
140 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
154 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
178 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
202 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
217 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
230 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
[all …]
|
| D | kernel_graph_test.cc | 41 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
64 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
101 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
112 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
143 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
190 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ |
| D | ge_backend_optimization.cc | 74 void GEBackendOptimization(const KernelGraphPtr &kernel_graph) { in GEBackendOptimization()
124 void GEBackendOptimizeACL(const KernelGraphPtr &kernel_graph) { in GEBackendOptimizeACL()
168 void GEBackendOptimizeACLAfterKernelSelect(const KernelGraphPtr &kernel_graph) { in GEBackendOptimizeACLAfterKernelSelect()
214 void GEUnifyMindIR(const KernelGraphPtr &kernel_graph) { in GEUnifyMindIR()
239 void GEAfterInlineOptimize(const KernelGraphPtr &kernel_graph) { in GEAfterInlineOptimize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/ |
| D | gpu_session.cc | 145 void GPUSession::Optimize(const std::shared_ptr<KernelGraph> &kernel_graph) { in Optimize()
174 void GPUSession::HardwareOptimize(const std::shared_ptr<KernelGraph> &kernel_graph) { in HardwareOptimize()
202 void GPUSession::RunOpOptimize(const std::shared_ptr<KernelGraph> &kernel_graph) { in RunOpOptimize()
212 void GPUSession::RunOpHardwareOptimize(const std::shared_ptr<KernelGraph> &kernel_graph) { in RunOpHardwareOptimize()
222 void GPUSession::GraphKernelOptimize(const std::shared_ptr<KernelGraph> &kernel_graph) { in GraphKernelOptimize()
230 void GPUSession::AssignStream(const std::shared_ptr<KernelGraph> &kernel_graph) { in AssignStream()
248 const KernelGraph *kernel_graph, bool is_gradient_out) const { in RunOpAllocateMemory() argument
272 bool UpdatedByAssign(const KernelGraphPtr &kernel_graph, const AnfNodePtr &node) { in UpdatedByAssign()
335 void GPUSession::LoadInputData(const std::shared_ptr<KernelGraph> &kernel_graph, in LoadInputData()
497 void GPUSession::PreExecuteGraph(const std::shared_ptr<KernelGraph> &kernel_graph, in PreExecuteGraph()
[all …]
|
| D | gpu_inference_session.cc | 30 void GpuInferenceSession::LoadInputData(const std::shared_ptr<KernelGraph> &kernel_graph, in LoadInputData()
65 auto kernel_graph = GetGraph(graph_id); in CompileGraphImpl() local
100 auto kernel_graph = GetGraph(graph_id); in CheckModelInputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/ |
| D | inline_control_flow_scheduler.cc | 38 const auto &kernel_graph = func_graph->cast<KernelGraphPtr>(); in IsInlineKernelActor() local
52 const auto &kernel_graph = func_graph->cast<KernelGraphPtr>(); in GetBranchNameByKernelActor() local
173 … const KernelGraphPtr &kernel_graph) { in GetBranchNameByConditionGatherActor()
221 ConditionSwitchActor *const condition_switch_actor, const KernelGraphPtr &kernel_graph) { in InitOutputDataBranchInfoForConditionSwitchActor()
285 ConditionSwitchActor *const condition_switch_actor, const KernelGraphPtr &kernel_graph) { in InitOutputControlBranchInfoForConditionSwitchActor()
331 ConditionSwitchActor *const condition_switch_actor, const KernelGraphPtr &kernel_graph) { in InitOutputBranchInfoForConditionSwitchActor()
354 const auto &kernel_graph = graph->cast<KernelGraphPtr>(); in HandleConditionSwitchActor() local
409 const KernelGraph *const kernel_graph) { in FixRefCountForRefNode()
480 const auto &kernel_graph = dynamic_cast<KernelGraph *>(graph.get()); in FixRefCountForInputNode() local
510 … const KernelGraphPtr &kernel_graph) { in FixRefCountByConditionGatherActor()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/ |
| D | gpu_stream_assign.cc | 34 void AssignGpuStream(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in AssignGpuStream()
81 void AssignDefaultGpuStream(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in AssignDefaultGpuStream()
89 bool FindAllReduceStreamSwitchPos(const std::shared_ptr<session::KernelGraph> &kernel_graph, in FindAllReduceStreamSwitchPos()
174 void InsertStreamSwitchNode(const std::shared_ptr<session::KernelGraph> &kernel_graph, in InsertStreamSwitchNode()
206 bool GenSendRecvCNodesForAllReduce(const std::shared_ptr<session::KernelGraph> &kernel_graph, in GenSendRecvCNodesForAllReduce()
228 CNodePtr CreateStreamSwitchNode(const std::shared_ptr<session::KernelGraph> &kernel_graph, const st… in CreateStreamSwitchNode()
248 void CacheSendRecvCNodesForAllReduce(const std::shared_ptr<session::KernelGraph> &kernel_graph, in CacheSendRecvCNodesForAllReduce()
|
| D | gpu_launch_mul.cc | 31 void GPULaunchMul::KernelSelect(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelSelect()
35 void GPULaunchMul::KernelBuild(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelBuild()
|
| D | gpu_launch_kernel.cc | 49 void GPULaunchkernel::KernelSelect(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelSelect()
58 void GPULaunchkernel::KernelBuild(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelBuild()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/ |
| D | cpu_session.cc | 94 void CPUSession::Optimize(const std::shared_ptr<KernelGraph> &kernel_graph) { in Optimize()
164 auto kernel_graph = GetGraph(graph_id); in CreateOutputTensors() local
169 void CPUSession::LoadInputData(const std::shared_ptr<KernelGraph> &kernel_graph, in LoadInputData()
201 void CPUSession::PreExecuteGraph(const std::shared_ptr<KernelGraph> &kernel_graph, in PreExecuteGraph()
207 void CPUSession::PostExecuteGraph(const std::shared_ptr<KernelGraph> &kernel_graph, in PostExecuteGraph()
214 void CPUSession::ExecuteGraph(const std::shared_ptr<KernelGraph> &kernel_graph) { in ExecuteGraph()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/ |
| D | graph_kernel_optimization.cc | 327 void GraphKernelOptimizer::Run(const KernelGraphPtr &kernel_graph) { in Run()
372 void GraphKernelOptimizer::RunKernelPacket(const KernelGraphPtr &kernel_graph) { in RunKernelPacket()
378 void GraphKernelOptimize(const KernelGraphPtr &kernel_graph) { in GraphKernelOptimize()
383 void KernelPacketOptimize(const KernelGraphPtr &kernel_graph) { in KernelPacketOptimize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/ |
| D | insert_cast.cc | 44 auto kernel_graph = func_graph->cast<KernelGraphPtr>(); in InsertCastForMultipleOutput() local
90 auto kernel_graph = func_graph->cast<KernelGraphPtr>(); in InsertCastForOutput() local
176 auto kernel_graph = func_graph->cast<std::shared_ptr<session::KernelGraph>>(); in InsertCastForInput() local
206 auto kernel_graph = func_graph->cast<std::shared_ptr<session::KernelGraph>>(); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/ |
| D | convert_const_scalar_to_tensor.cc | 30 AnfNodePtr ConvertTensorInput(const KernelGraphPtr &kernel_graph, const AnfNodePtr &input_node) { in ConvertTensorInput()
88 auto kernel_graph = func_graph->cast<KernelGraphPtr>(); in Process() local
|
| D | insert_tensor_move_for_communication.cc | 26 const KernelGraphPtr &kernel_graph) { in IsNeedInsertForInput()
52 auto kernel_graph = std::dynamic_pointer_cast<session::KernelGraph>(graph); in Run() local
|
| D | convert_tuple_output_to_maketuple.cc | 39 auto kernel_graph = graph->cast<KernelGraphPtr>(); in ConvertTupleInputToMakeTuple() local
115 auto kernel_graph = graph->cast<KernelGraphPtr>(); in Process() local
|
| D | eliminate_func_data_type.cc | 46 void RemoveInputFuncNodeForKernelGraph(const KernelGraphPtr &kernel_graph, const AnfNodePtr &func_i… in RemoveInputFuncNodeForKernelGraph()
101 auto kernel_graph = func_graph->cast<KernelGraphPtr>(); in Process() local
|
| D | add_dynamic_shape_attr.cc | 31 auto kernel_graph = func_graph->cast<KernelGraphPtr>(); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | session_basic.h | 203 virtual void PreExecuteGraph(const std::shared_ptr<KernelGraph> &kernel_graph, in PreExecuteGraph()
210 virtual void PostExecuteGraph(const std::shared_ptr<KernelGraph> &kernel_graph, in PostExecuteGraph()
217 …virtual void ExecuteGraph(const std::shared_ptr<KernelGraph> &kernel_graph) { MS_EXCEPTION_IF_NULL… in ExecuteGraph()
230 auto kernel_graph = GetGraph(graph_id); in LoadInputs() local
239 virtual void LoadInputData(const std::shared_ptr<KernelGraph> &kernel_graph, in LoadInputData()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ |
| D | acl_stream_assign.cc | 38 void SetForSwitchInline(const NotNull<KernelGraphPtr> &kernel_graph, const CNodePtr &send_cnode, in SetForSwitchInline()
100 void AclStreamAssign::AssignStream(const NotNull<KernelGraphPtr> &kernel_graph) { in AssignStream()
159 const NotNull<KernelGraphPtr> &kernel_graph, in GenKernelIoExecInfoMap()
212 void AclStreamAssign::AddBoundarySendRecvKernel(const NotNull<KernelGraphPtr> &kernel_graph, uint32… in AddBoundarySendRecvKernel()
230 void AclStreamAssign::ProcessSideEffect(const NotNull<KernelGraphPtr> &kernel_graph, const CNodePtr… in ProcessSideEffect()
277 const NotNull<KernelGraphPtr> &kernel_graph, in UpdateEventsToExecutionOrder()
351 void AclStreamAssign::ProcessStreamForInputs(const NotNull<KernelGraphPtr> &kernel_graph, const CNo… in ProcessStreamForInputs()
384 void AclStreamAssign::InsertEventsForOutputs(const NotNull<KernelGraphPtr> &kernel_graph, const CNo… in InsertEventsForOutputs()
445 void AclStreamAssign::InsertEvents(const NotNull<KernelGraphPtr> &kernel_graph, const CNodePtr &par… in InsertEvents()
502 void AclStreamAssign::InsertEventsForSendOp(const NotNull<KernelGraphPtr> &kernel_graph, const CNod… in InsertEventsForSendOp()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/ |
| D | cpu_kernel_runtime.cc | 63 void CPUKernelRuntime::AssignKernelGraphAddress(const session::KernelGraph *kernel_graph) { in AssignKernelGraphAddress()
88 void CPUKernelRuntime::AssignValueNodeAddress(const session::KernelGraph *kernel_graph) { in AssignValueNodeAddress()
187 tensor::TensorPtr CPUKernelRuntime::CreateTensorForOutput(session::KernelGraph *kernel_graph, const… in CreateTensorForOutput()
244 session::KernelGraph *kernel_graph, const session::KernelWithIndex &kernel_with_index, in GetOrCreateTensorForOutput()
282 void CPUKernelRuntime::CreateOutputTensors(session::KernelGraph *kernel_graph, in CreateOutputTensors()
311 void CPUKernelRuntime::BindInputTensorAddressPtr(const session::KernelGraph &kernel_graph, in BindInputTensorAddressPtr()
386 void CPUKernelRuntime::BindInputOutput(session::KernelGraph *kernel_graph, const std::vector<tensor… in BindInputOutput()
442 bool CPUKernelRuntime::Run(const session::KernelGraph &kernel_graph, bool) { in Run()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/runtime/device/gsm/ |
| D | swap_strategy_builder_test.cc | 55 auto kernel_graph = kernel_graph_add_net_; in TEST_F() local
85 auto kernel_graph = kernel_graph_add_net_; in TEST_F() local
130 auto kernel_graph = kernel_graph_add_with_all_reduce_net_; in TEST_F() local
|
| D | mem_usage_analyzer_test.cc | 43 auto kernel_graph = Compile(func_graph); in TEST_F() local
74 auto kernel_graph = Compile(func_graph); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/ |
| D | insert_identity.cc | 83 auto kernel_graph = func_graph->cast<KernelGraphPtr>(); in InsertTransIdentityForInput() local
105 auto kernel_graph = func_graph->cast<KernelGraphPtr>(); in InsertCastIdentityForInput() local
125 auto kernel_graph = func_graph->cast<KernelGraphPtr>(); in InsertIdentityForOutput() local
|
| D | ge_convert_const_input_to_tensor_input.cc | 50 auto kernel_graph = func_graph->cast<std::shared_ptr<session::KernelGraph>>(); in UpdateConstInput() local
72 auto kernel_graph = func_graph->cast<std::shared_ptr<session::KernelGraph>>(); in ConstInputToTensorInput() local
|