Home
last modified time | relevance | path

Searched defs:GenDynamicKernel (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_kernel_runtime.h47 bool GenDynamicKernel(const session::KernelGraph &graph) override { return true; } in GenDynamicKernel() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Ddynamic_shape_kernel.cc52 device::DynamicKernelPtr DynamicShapeKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *s… in GenDynamicKernel() function in mindspore::kernel::DynamicShapeKernelMod
Ddynamic_broadcast_gradient_args_kernel.cc213 device::DynamicKernelPtr DynamicBroadcastGradientArgsKernelMod::GenDynamicKernel(const CNodePtr &cn… in GenDynamicKernel() function in mindspore::kernel::DynamicBroadcastGradientArgsKernelMod
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Dprofiling_kernel_mod.cc68 device::DynamicKernelPtr ProfilingKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stre… in GenDynamicKernel() function in mindspore::kernel::ProfilingKernelMod
Dmemcpy_async.cc135 device::DynamicKernelPtr MemCpyAsyncKernel::GenDynamicKernel(const CNodePtr &cnode_ptr, void *strea… in GenDynamicKernel() function in mindspore::kernel::MemCpyAsyncKernel
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.h48 bool GenDynamicKernel(const session::KernelGraph &graph) override { return true; } in GenDynamicKernel() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_mod.cc206 device::DynamicKernelPtr AicpuOpKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream… in GenDynamicKernel() function in mindspore::kernel::AicpuOpKernelMod
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhccl_kernel.cc275 device::DynamicKernelPtr HcclKernel::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) { in GenDynamicKernel() function in mindspore::kernel::HcclKernel
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_mod.cc113 device::DynamicKernelPtr TbeKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr… in GenDynamicKernel() function in mindspore::kernel::TbeKernelMod
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel.h184 …virtual device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) { r… in GenDynamicKernel() function
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.cc429 bool AscendKernelRuntime::GenDynamicKernel(const session::KernelGraph &graph) { in GenDynamicKernel() function in mindspore::device::ascend::AscendKernelRuntime