Home
last modified time | relevance | path

Searched defs:kernel_inputs (Results 1 – 16 of 16) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/
Dhccl_dynamic_kernel.cc45 AddressPtrList kernel_inputs; in UpdateArgs() local
83 AddressPtrList kernel_inputs; in StaticShapeExecute() local
Dai_core_dynamic_kernel.cc110 AddressPtrList kernel_inputs; in UpdateArgs() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dlaunch_kernel.cc42 std::vector<kernel::AddressPtr> kernel_inputs; in ObtainKernelInputs() local
80 auto kernel_inputs = ObtainKernelInputs(kernel_mod_->GetInputSizeList(), inputs_addr); in LaunchSingleKernel() local
Dkernel_runtime.cc1121AddressPtrList *kernel_inputs, AddressPtrList *const kernel_workspaces, in GenLaunchArgs()
1181 void KernelRuntime::GenAddrCleanLaunchArgs(const CNodePtr &cnode, AddressPtrList *kernel_inputs, in GenAddrCleanLaunchArgs()
1299AddressPtrList *kernel_inputs, AddressPtrList *kernel_workspaces, in AssignKernelAddress()
1418 AddressPtrList kernel_inputs; in LaunchKernel() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/
Dtask_generator.cc60 …kGenerator::LaunchAddrCleanAkgKernel(const CNodePtr &anf_node_ptr, AddressPtrList *kernel_inputs) { in LaunchAddrCleanAkgKernel()
95 …TaskGenerator::LaunchAddrCleanKernel(const CNodePtr &anf_node_ptr, AddressPtrList *kernel_inputs) { in LaunchAddrCleanKernel()
148 AddressPtrList kernel_inputs; in LaunchKernel() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_transdata.cc56 std::vector<kernel::AddressPtr> kernel_inputs; in LaunchOpKernel() local
Dascend_launch_atomic_clean.cc54 std::vector<kernel::AddressPtr> kernel_inputs; in LaunchOpKernel() local
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/
Dascend_ps_cache.cc236 AddressPtrList kernel_inputs; in HashSwapOut() local
277 AddressPtrList kernel_inputs; in HashSwapIn() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Dmemcpy_async.cc136 AddressPtrList kernel_inputs; in GenDynamicKernel() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dcommon.cc76 std::vector<Tensor *> kernel_inputs; in TestMain() local
242 std::vector<Tensor *> kernel_inputs; in TestMain() local
/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Ddebugger_utils.cc57 auto kernel_inputs = launch_info_->inputs_; in LoadInputs() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc135 const std::vector<mindspore::kernel::AddressPtr> &kernel_inputs, in LoadKernelData()
773 AddressPtrList kernel_inputs; in LaunchKernelDynamic() local
893 AddressPtrList kernel_inputs; in RunOpLaunchKernelDynamic() local
1081 … const mindspore::AnfNodePtr &kernel, AddressPtrList *kernel_inputs, in AllocKernelDynamicRes()
1096 …ime::AllocKernelInputDynamicRes(const mindspore::AnfNodePtr &kernel, AddressPtrList *kernel_inputs, in AllocKernelInputDynamicRes()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_mod.cc207 AddressPtrList kernel_inputs; in GenDynamicKernel() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_mod.cc114 AddressPtrList kernel_inputs; in GenDynamicKernel() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_kernel_runtime.cc434 std::vector<kernel::AddressPtr> kernel_inputs; in Run() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc2379 std::vector<AddressPtr> kernel_inputs; in CacheAddrForKernel() local
2437 std::vector<AddressPtr> kernel_inputs; in CacheAddrForAtomicClean() local