Home
last modified time | relevance | path

Searched defs:kernel_outputs (Results 1 – 15 of 15) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/
Dhccl_dynamic_kernel.cc47 AddressPtrList kernel_outputs; in UpdateArgs() local
85 AddressPtrList kernel_outputs; in StaticShapeExecute() local
Dai_core_dynamic_kernel.cc112 AddressPtrList kernel_outputs; in UpdateArgs() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dlaunch_kernel.cc62 auto kernel_outputs = ObtainKernelAddress(outputs_list, &outputs_addr_); in ObtainKernelOutputs() local
82 auto kernel_outputs = ObtainKernelOutputs(kernel_mod_->GetOutputSizeList()); in LaunchSingleKernel() local
Dkernel_runtime.cc1122 AddressPtrList *kernel_outputs) { in GenLaunchArgs()
1300 AddressPtrList *kernel_outputs) { in AssignKernelAddress()
1420 AddressPtrList kernel_outputs; in LaunchKernel() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_transdata.cc64 auto kernel_outputs = ObtainKernelOutputs(kernel_mod_->GetOutputSizeList()); in LaunchOpKernel() local
Dascend_launch_atomic_clean.cc62 auto kernel_outputs = ObtainKernelOutputs(kernel_mod_->GetOutputSizeList()); in LaunchOpKernel() local
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/
Dascend_ps_cache.cc237 AddressPtrList kernel_outputs = { in HashSwapOut() local
278 AddressPtrList kernel_outputs; in HashSwapIn() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Dmemcpy_async.cc138 AddressPtrList kernel_outputs; in GenDynamicKernel() local
/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Ddebugger_utils.cc84 auto kernel_outputs = launch_info_->outputs_; in LoadOutputs() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc137const std::vector<mindspore::kernel::AddressPtr> &kernel_outputs, int exec_order, void *stream_ptr, in LoadKernelData()
775 AddressPtrList kernel_outputs; in LaunchKernelDynamic() local
895 AddressPtrList kernel_outputs; in RunOpLaunchKernelDynamic() local
1082 … AddressPtrList *kernel_workspaces, AddressPtrList *kernel_outputs, in AllocKernelDynamicRes()
1136 … const mindspore::AnfNodePtr &kernel, AddressPtrList *kernel_outputs, in AllocKernelOutputDynamicRes()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_mod.cc209 AddressPtrList kernel_outputs; in GenDynamicKernel() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_mod.cc116 AddressPtrList kernel_outputs; in GenDynamicKernel() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/
Dtask_generator.cc150 AddressPtrList kernel_outputs; in LaunchKernel() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_kernel_runtime.cc436 std::vector<kernel::AddressPtr> kernel_outputs; in Run() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc2381 std::vector<AddressPtr> kernel_outputs; in CacheAddrForKernel() local