| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/rl/ |
| D | reservoir_replay_buffer_cpu_kernel.cc | 34 bool ReservoirReplayBufferCreateCpuKernel::Init(const std::vector<KernelTensor *> &, in Init()
63 bool ReservoirReplayBufferCreateCpuKernel::Launch(const std::vector<KernelTensor *> &, in Launch()
72 bool ReservoirReplayBufferPushCpuKernel::Init(const std::vector<KernelTensor *> &inputs, in Init()
82 bool ReservoirReplayBufferPushCpuKernel::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
94 bool ReservoirReplayBufferSampleCpuKernel::Init(const std::vector<KernelTensor *> &, in Init()
112 bool ReservoirReplayBufferSampleCpuKernel::Launch(const std::vector<KernelTensor *> &, in Launch()
118 bool ReservoirReplayBufferDestroyCpuKernel::Init(const std::vector<KernelTensor *> &, in Init()
125 bool ReservoirReplayBufferDestroyCpuKernel::Launch(const std::vector<KernelTensor *> &, in Launch()
|
| D | tensors_queue_cpu_kernel.cc | 54 bool TensorsQueueCreateCpuKernelMod::Launch(const std::vector<KernelTensor *> &, const std::vector<… in Launch()
87 bool TensorsQueuePutCpuKernelMod::Launch(const std::vector<KernelTensor *> &inputs, const std::vect… in Launch()
143 bool TensorsQueueGetCpuKernelMod::Launch(const std::vector<KernelTensor *> &inputs, const std::vect… in Launch()
186 bool TensorsQueueClearCpuKernelMod::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
208 bool TensorsQueueSizeCpuKernelMod::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
233 bool TensorsQueueCloseCpuKernelMod::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
|
| D | priority_replay_buffer_cpu_kernel.cc | 69 bool PriorityReplayBufferCreateCpuKernel::Launch(const std::vector<KernelTensor *> &, in Launch()
89 bool PriorityReplayBufferPushCpuKernel::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
121 bool PriorityReplayBufferSampleCpuKernel::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
170 bool PriorityReplayBufferUpdateCpuKernel::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
200 bool PriorityReplayBufferDestroyCpuKernel::Launch(const std::vector<KernelTensor *> &, in Launch()
|
| D | tensor_array_clear_kernel.cc | 37 bool TensorArrayClearCpuKernelMod::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
|
| D | tensor_array_close_kernel.cc | 38 bool TensorArrayCloseCpuKernelMod::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/control/ |
| D | recv_gpu_kernel.h | 33 bool Launch(const std::vector<KernelTensor *> &, const std::vector<KernelTensor *> &, in Launch()
40 bool Init(const std::vector<KernelTensor *> &, const std::vector<KernelTensor *> &) override { in Init()
|
| D | send_gpu_kernel.h | 33 bool Launch(const std::vector<KernelTensor *> &, const std::vector<KernelTensor *> &, in Launch()
40 bool Init(const std::vector<KernelTensor *> &, const std::vector<KernelTensor *> &) override { in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/rpc/ |
| D | rpc_send_kernel.h | 33 bool Launch(const std::vector<KernelTensor *> &, const std::vector<KernelTensor *> &, in Launch()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/ |
| D | hcom_barrier.cc | 22 bool HcomBarrierKernel::Launch(const std::vector<KernelTensor *> &, const std::vector<KernelTensor … in Launch()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/environ/ |
| D | environ_gpu_destroy_all.cc | 36 bool EnvironDestroyAllGpuKernelMod::Launch(const std::vector<KernelTensor *> &, const std::vector<K… in Launch()
|
| D | environ_gpu_create.cc | 39 bool EnvironCreateGpuKernelMod::Launch(const std::vector<KernelTensor *> &, const std::vector<Kerne… in Launch()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/environ/ |
| D | environ_cpu_destroy_all.cc | 38 bool EnvironDestroyAllCpuKernelMod::Launch(const std::vector<KernelTensor *> &, const std::vector<K… in Launch()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/rts/ |
| D | condition_gather.cc | 33 bool ConditionGatherKernel::Launch(const std::vector<KernelTensor *> &, const std::vector<KernelTen… in Launch()
|
| D | send.cc | 44 bool SendKernel::Launch(const std::vector<KernelTensor *> &, const std::vector<KernelTensor *> &, in Launch()
|
| D | condition_switch.cc | 33 bool ConditionSwitchKernel::Launch(const std::vector<KernelTensor *> &, const std::vector<KernelTen… in Launch()
|
| D | recv.cc | 47 bool RecvKernel::Launch(const std::vector<KernelTensor *> &, const std::vector<KernelTensor *> &, in Launch()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/rl/ |
| D | tensors_queue_gpu_kernel.cc | 52 bool TensorsQueueCreateKernelMod::Launch(const std::vector<KernelTensor *> &, const std::vector<Ker… in Launch()
87 bool TensorsQueuePutKernelMod::Launch(const std::vector<KernelTensor *> &inputs, const std::vector<… in Launch()
140 bool TensorsQueueGetKernelMod::Launch(const std::vector<KernelTensor *> &inputs, const std::vector<… in Launch()
183 bool TensorsQueueClearKernelMod::Launch(const std::vector<KernelTensor *> &inputs, const std::vecto… in Launch()
204 bool TensorsQueueSizeKernelMod::Launch(const std::vector<KernelTensor *> &inputs, const std::vector… in Launch()
227 bool TensorsQueueCloseKernelMod::Launch(const std::vector<KernelTensor *> &inputs, const std::vecto… in Launch()
|
| D | reservoir_replay_buffer_gpu_kernel.cc | 74 bool ReservoirReplayBufferCreateGpuKernel::Launch(const std::vector<KernelTensor *> &, in Launch()
106 bool ReservoirReplayBufferPushGpuKernel::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
140 bool ReservoirReplayBufferSampleGpuKernel::Launch(const std::vector<KernelTensor *> &, in Launch()
158 bool ReservoirReplayBufferDestroyGpuKernel::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
|
| D | priority_replay_buffer_gpu_kernel.cc | 75 bool PriorityReplayBufferCreateGpuKernel::Launch(const std::vector<KernelTensor *> &, in Launch()
110 bool PriorityReplayBufferPushGpuKernel::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
145 bool PriorityReplayBufferSampleGpuKernel::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
191 bool PriorityReplayBufferUpdateGpuKernel::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
218 bool PriorityReplayBufferDestroyGpuKernel::Launch(const std::vector<KernelTensor *> &inputs, in Launch()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/data/ |
| D | dataset_init_kernel.cc | 52 bool DatasetInitKernelMod::Launch(const std::vector<KernelTensor *> &, const std::vector<KernelTens… in Launch()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | memcpy_gpu_kernel.h | 35 …bool Init(const std::vector<KernelTensor *> &, const std::vector<KernelTensor *> &) override { ret… in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | standard_laplace_cpu_kernel.cc | 28 bool StandardLaplaceCpuKernelMod::Init(const std::vector<KernelTensor *> &, const std::vector<Kerne… in Init()
36 bool StandardLaplaceCpuKernelMod::Launch(const std::vector<kernel::KernelTensor *> &inputs, in Launch()
|
| D | convert_to_dynamic_cpu_kernel.cc | 27 bool ConvertToDynamicCpuKernelMod::Init(const std::vector<KernelTensor *> &, const std::vector<Kern… in Init()
61 bool ConvertToDynamicCpuKernelMod::Launch(const std::vector<kernel::KernelTensor *> &inputs, in Launch()
|
| D | memcpy_cpu_kernel.h | 34 …bool Init(const std::vector<KernelTensor *> &, const std::vector<KernelTensor *> &) override { ret… in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_akg/ |
| D | dynamic_akg_cpu_kernel_mod.h | 46 …bool Init(const std::vector<KernelTensor *> &, const std::vector<KernelTensor *> &) override { ret… in Init()
|