Home
last modified time | relevance | path

Searched refs:GenDynamicKernel (Results 1 – 22 of 22) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Ddynamic_shape_kernel.h37 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
Ddynamic_broadcast_gradient_args_kernel.h38 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
Ddynamic_shape_kernel.cc52 device::DynamicKernelPtr DynamicShapeKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *s… in GenDynamicKernel() function in mindspore::kernel::DynamicShapeKernelMod
Dhost_kernel_mod.h37 …device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override = …
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.h30 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
Dmemcpy_async.h37 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
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/backend/kernel_compiler/tbe/
Dtbe_kernel_mod.h45 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
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/aicpu/
Daicpu_kernel_mod.h34 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
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/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/hccl/
Dhccl_kernel.h44 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
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/runtime/device/ascend/
Dascend_kernel_runtime.h51 bool GenDynamicKernel(const session::KernelGraph &graph) override;
Dascend_kernel_runtime.cc429 bool AscendKernelRuntime::GenDynamicKernel(const session::KernelGraph &graph) { in GenDynamicKernel() function in mindspore::device::ascend::AscendKernelRuntime
438 auto dynamic_kernel = kernel_mod->GenDynamicKernel(cnode, stream_); in GenDynamicKernel()
462 return GenDynamicKernel(graph); in GenTask()
/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/gpu/
Dgpu_kernel_runtime.h48 bool GenDynamicKernel(const session::KernelGraph &graph) override { return true; } in GenDynamicKernel() function
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.h70 virtual bool GenDynamicKernel(const session::KernelGraph &graph) = 0;
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_session.cc1335 if (!runtime_instance->GenDynamicKernel(*kernel_graph)) { in BuildDynamicKernel()