| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/acl/ |
| D | acl_kernel_lib.cc | 39 …std::shared_ptr<kernel::KernelMod> kernel_mod = kernel::Factory<kernel::KernelMod>::Instance().Cre… in CreateKernelMod() local
56 …auto kernel_mod = AclKernelLib::CreateKernelMod(spec.op_type, spec.attr, spec.format, spec.backend… in CreateKernel() local
|
| D | acl_lite_kernel.cc | 27 AclLiteKernel::AclLiteKernel(std::shared_ptr<mindspore::kernel::KernelMod> kernel_mod, BaseOperator… in AclLiteKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/default/ |
| D | default_kernel_lib.cc | 34 auto kernel_mod = Factory<NativeCpuKernelMod>::Instance().Create(op_type.TypeName()); in CreateKernelMod() local
54 …auto kernel_mod = DefaultKernelLib::CreateKernelMod(spec.op_type, spec.attr, spec.format, spec.bac… in CreateKernel() local
|
| D | kernel_mod_kernel.h | 30 …KernelModKernel(std::shared_ptr<mindspore::kernel::KernelMod> kernel_mod, BaseOperatorPtr base_ope… in KernelModKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/ |
| D | gpu_kernel_runtime.cc | 356 auto kernel_mod = AnfAlgo::GetKernelMod(item[0]); in AllocInplaceNodeMemory() local
636 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in InitKernelOutputAddress() local
655 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in InitKernelWorkspaceAddress() local
698 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in ClearKernelOutputAddress() local
720 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in ClearKernelWorkspaceAddress() local
771 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelDynamic() local
881 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelWithoutMock() local
908 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in RunOpLaunchKernelDynamic() local
946 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelWithTimeProfiling() local
1114 bool GPUKernelRuntime::AllocKernelDynamicRes(const mindspore::kernel::KernelMod &kernel_mod, in AllocKernelDynamicRes()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/ |
| D | cpu_simple_mem_plan.cc | 52 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in MemPlan() local
98 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in MemAssign() local
|
| D | cpu_memory_manager.cc | 185 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in IncreaseAddressRefCount() local
210 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in DecreaseAddressRefCount() local
|
| D | cpu_kernel_runtime.cc | 160 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in AssignKernelOutputAddress() local
433 auto kernel_mod = AnfAlgo::GetKernelMod(node); in GetRuntimeAddressFromNode() local
461 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | kernel_mod_cache.h | 37 …void SetCache(const std::string &key, const KernelModPtr &kernel_mod) { kernel_mod_cache_[key] = k… in SetCache()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/ |
| D | op_runner.cc | 275 auto kernel_mod = AnfAlgo::GetKernelMod(node); in MallocForKernelInput() local
312 auto kernel_mod = AnfAlgo::GetKernelMod(node); in MallocForKernelOutput() local
407 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in GetWorkspaceKernelTensors() local
459 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in GetWorkspaceKernelTensorsDynamic() local
511 auto *kernel_mod = AnfAlgo::GetKernelMod(kernel); in InferNodeRealShape() local
520 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in ResizeKernelMod() local
605 auto kernel_mod = runtime_info->GetKernelMod(); in LaunchKernels() local
619 auto kernel_mod = AnfAlgo::GetKernelMod(node); in LaunchKernels() local
654 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in UpdateOutputDeviceInfo() local
912 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in RunSingleOpGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/mindir_loader/mindir_model/ |
| D | kernel_mod_util.cc | 31 std::shared_ptr<kernel::KernelMod> kernel_mod = nullptr; in GetInnerKernel() local
|
| D | inner_kernel.h | 40 InnerKernel(std::shared_ptr<mindspore::kernel::KernelMod> kernel_mod, in InnerKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 243 const auto kernel_mod = AnfAlgo::GetKernelMod(node); in GetCommunicationOutputInfo() local
290 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in ResetNodeAddress() local
366 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in RunOpClearMemory() local
458 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in RunOpAssignOutputMemory() local
510 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in RunOpAssignWorkSpaceMemory() local
780 auto kernel_mod = AnfAlgo::GetKernelMod(node); in AssignCommunicationNodeOutputMem() local
953 auto kernel_mod = AnfAlgo::GetKernelMod(node); in AssignNodeOutputMem() local
1158 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in MemSchedulerPreCompute() local
1243 auto kernel_mod = AnfAlgo::GetKernelMod(node); in AssignWorkSpaceMem() local
1257 void KernelRuntime::GenLaunchArgs(const mindspore::kernel::KernelMod &kernel_mod, const mindspore::… in GenLaunchArgs()
[all …]
|
| D | kernel_info.cc | 190 void KernelInfo::set_kernel_mod(const kernel::KernelModPtr &kernel_mod) { kernel_mod_ = kernel_mod;… in set_kernel_mod()
196 const kernel::KernelMod *KernelInfo::kernel_mod() const { return kernel_mod_.get(); } in kernel_mod() function in mindspore::device::KernelInfo
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/ |
| D | convert_call_to_prim.cc | 56 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/mem_reuse/ |
| D | mem_reuse.cc | 40 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_cnode); in InitDynamicOutputKernelRef() local
105 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_cnode); in InitDynamicWorkspaceKernelRef() local
216 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in SetOutputMap() local
231 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in SetWkMap() local
|
| D | mem_reuse_checker.cc | 120 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CalculOriDy() local
136 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CalculOriWk() local
237 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_input); in CheckGraphOutputAssigned() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/ |
| D | pyboost_utils.cc | 157 auto kernel_mod = cache_helper.GetKernelMod(key); in CreateKernelMod() local
319 …ce::DeviceAddressPtrList PyBoostUtils::CreateWorkSpaceDeviceAddress(const KernelModPtr &kernel_mod, in CreateWorkSpaceDeviceAddress()
361 …auto kernel_mod = PyBoostUtils::CreateKernelMod(primitive, real_name, device_context, input_addres… in LaunchKernel() local
519 const std::string &op_name, const kernel::KernelModPtr &kernel_mod, in GetKernelAttr()
563 const auto &kernel_mod = device_context->GetKernelExecutor(false)->CreateKernelMod(op_name); in SelectKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/bisheng/ |
| D | bisheng_kernel_build.cc | 71 auto kernel_mod = Factory<BiShengKernelMod>::Instance().Create(kernel_name); in BiShengOpBuild() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | data_source_actor.cc | 107 const auto &kernel_mod = kernel_info_->MutableKernelMod(); in FillDataBuffer() local
189 auto kernel_mod = AnfAlgo::GetKernelMod(data_kernel_); in OnMemoryAllocFinish() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/session/ |
| D | memory_offload_session.cc | 39 …std::shared_ptr<kernel::KernelMod> kernel_mod = kernel::Factory<kernel::KernelMod>::Instance().Cre… in BuildCustomAscendKernelImpl() local
|
| D | single_op_session.cc | 140 …std::shared_ptr<kernel::KernelMod> kernel_mod = kernel::Factory<kernel::KernelMod>::Instance().Cre… in BuildCustomAscendKernelImpl() local
213 kernel::KernelModPtr kernel_mod; in BuildCustomAscendKernel() local
551 …std::shared_ptr<kernel::KernelMod> kernel_mod = kernel::Factory<kernel::KernelMod>::Instance().Cre… in AscendFinalize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/ |
| D | runtime_utils.cc | 102 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_node); in UpdateKernelNodeOutputInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/debugger/ |
| D | debugger_utils.cc | 75 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in GetValidDumpIndex() local
104 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in LoadInputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/ |
| D | memory_swap_node_scheduler.cc | 212 auto kernel_mod = AnfAlgo::GetKernelMod(node); in CollectKernelInputMemStatistic() local
250 auto kernel_mod = AnfAlgo::GetKernelMod(node); in CollectKernelOutputMemStatistic() local
281 auto kernel_mod = AnfAlgo::GetKernelMod(node); in CollectKernelWorkspaceMemStatistic() local
|