Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Dlaunch_kernel.cc43 std::vector<kernel::KernelTensor *> kernel_inputs; in ObtainKernelInputs() local
86 auto kernel_inputs = ObtainKernelInputs(input_size_list, inputs_addr); in LaunchSingleKernel() local
Dkernel_runtime.cc1361 …::GenKernelTensorLaunchArgs(const CNodePtr &cnode, std::vector<kernel::KernelTensor *> *kernel_inp… in GenKernelTensorLaunchArgs()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/acl/
Dacl_lite_kernel.cc44 std::vector<KernelTensor *> kernel_inputs; in ReSize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dlaunch_transdata.cc135 std::vector<kernel::KernelTensor *> kernel_inputs; in LaunchOpKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dcommon.cc69 std::vector<Tensor *> kernel_inputs; in TestMain() local
230 std::vector<Tensor *> kernel_inputs; in TestMain() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dgpu_kernel_runtime.cc148 const std::vector<mindspore::kernel::KernelTensor *> &kernel_inputs, in LoadKernelData()
793 KernelTensorList kernel_inputs; in LaunchKernelDynamic() local
1115 … const mindspore::AnfNodePtr &kernel, KernelTensorList *kernel_inputs, in AllocKernelDynamicRes()
1130 …e::AllocKernelInputDynamicRes(const mindspore::AnfNodePtr &kernel, KernelTensorList *kernel_inputs, in AllocKernelInputDynamicRes()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dcpu_kernel_runtime.cc478 std::vector<kernel::KernelTensor *> kernel_inputs; in Run() local