Searched refs:DynamicKernelPtr (Results 1 – 23 of 23) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | dynamic_shape_kernel.h | 37 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
|
| D | dynamic_broadcast_gradient_args_kernel.h | 38 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
|
| D | dynamic_shape_kernel.cc | 52 device::DynamicKernelPtr DynamicShapeKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *s… in GenDynamicKernel()
|
| D | host_kernel_mod.h | 37 …device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override = …
|
| D | dynamic_broadcast_gradient_args_kernel.cc | 213 device::DynamicKernelPtr DynamicBroadcastGradientArgsKernelMod::GenDynamicKernel(const CNodePtr &cn… in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | profiling_kernel_mod.h | 30 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
|
| D | memcpy_async.h | 37 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
|
| D | profiling_kernel_mod.cc | 68 device::DynamicKernelPtr ProfilingKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stre… in GenDynamicKernel()
|
| D | memcpy_async.cc | 135 device::DynamicKernelPtr MemCpyAsyncKernel::GenDynamicKernel(const CNodePtr &cnode_ptr, void *strea… in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/executor/ |
| D | dynamic_kernel.h | 61 using DynamicKernelPtr = std::shared_ptr<DynamicKernel>; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_mod.cc | 113 device::DynamicKernelPtr TbeKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr… in GenDynamicKernel()
146 device::DynamicKernelPtr executor = nullptr; in GenDynamicKernel()
|
| D | tbe_kernel_mod.h | 45 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_mod.h | 34 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
|
| D | aicpu_kernel_mod.cc | 206 device::DynamicKernelPtr AicpuOpKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream… in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hccl_kernel.h | 44 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
|
| D | hccl_kernel.cc | 275 device::DynamicKernelPtr HcclKernel::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) { in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel.h | 90 device::DynamicKernelPtr DynamicKernel() const { return dynamic_kernel_; } in DynamicKernel()
316 device::DynamicKernelPtr dynamic_kernel_;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel.h | 184 …virtual device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) { r… in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.h | 192 std::map<uint32_t, std::vector<DynamicKernelPtr>> graph_dynamic_kernel_map_;
|
| D | kernel_runtime.cc | 1474 std::vector<DynamicKernelPtr> dynamic_kernel_list; in LaunchKernelMod()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/ |
| D | gpu_device_context.cc | 384 device::DynamicKernelPtr dynamic_kernel = gpu_kernel->DynamicKernel(); in UpdateDynamicShape()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 760 device::DynamicKernelPtr dynamic_kernel = nullptr; in LaunchKernelDynamic()
880 device::DynamicKernelPtr dynamic_kernel = nullptr; in RunOpLaunchKernelDynamic()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_kernel_runtime.cc | 432 std::vector<DynamicKernelPtr> dynamic_kernels; in GenDynamicKernel()
|