| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/ |
| D | nccl_collective_gpu_kernel.h | 54 const std::vector<AddressPtr> &outputs, void *stream_ptr) override { in Launch()
157 void *stream_ptr) { in LaunchAllReduce()
170 void *stream_ptr) { in LaunchAllGather()
183 void *stream_ptr) { in LaunchReduceScatter()
197 void *stream_ptr) { in LaunchBroadcast()
|
| D | sync_batch_norm_grad_gpu_kernel.h | 43 const std::vector<AddressPtr> &outputs, void *stream_ptr) override { in Launch()
175 void LaunchAllReduce(reduce_type *input_addr, reduce_type *output_addr, void *stream_ptr) { in LaunchAllReduce()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/ |
| D | fake_quant_perchannel_gpu_kernel.cc | 117 void *stream_ptr) { in CalFakeQuantize()
126 … const std::vector<AddressPtr> &outputs, void *stream_ptr) { in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_receive.cc | 23 const std::vector<AddressPtr> &outputs, void *stream_ptr) { in Launch()
|
| D | hcom_all_reduce.cc | 23 const std::vector<AddressPtr> &outputs, void *stream_ptr) { in Launch()
|
| D | hcom_all_gather.cc | 23 const std::vector<AddressPtr> &outputs, void *stream_ptr) { in Launch()
|
| D | hcom_all_broadcast.cc | 24 const std::vector<AddressPtr> &, void *stream_ptr) { in Launch()
|
| D | hcom_send.cc | 23 const std::vector<AddressPtr> &, void *stream_ptr) { in Launch()
|
| D | hcom_all_reduce_scatter.cc | 23 const std::vector<AddressPtr> &outputs, void *stream_ptr) { in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | cholesky_trsm_solve_gpu_kernel.h | 54 const std::vector<AddressPtr> &outputs, void *stream_ptr) override { in Launch()
129 const std::vector<AddressPtr> &outputs, void *stream_ptr) { in LaunchNonSplitMatrix()
163 const std::vector<AddressPtr> &outputs, void *stream_ptr) { in LaunchSplitMatrix()
|
| D | logical_not_gpu_kernel.h | 41 const std::vector<AddressPtr> &outputs, void *stream_ptr) override { in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | memcpy_async.cc | 40 const std::vector<AddressPtr> &outputs, void *stream_ptr) { in Launch()
135 …DynamicKernelPtr MemCpyAsyncKernel::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) { in GenDynamicKernel()
|
| D | assign.cc | 30 const std::vector<AddressPtr> & /*outputs*/, void *stream_ptr) { in Launch()
|
| D | send.cc | 44 const std::vector<AddressPtr> &, void *stream_ptr) { in Launch()
|
| D | recv.cc | 45 const std::vector<AddressPtr> &, void *stream_ptr) { in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | ctcloss_gpu_kernel.h | 70 const std::vector<AddressPtr> &outputs, void *stream_ptr) override { in Launch()
152 const std::vector<AddressPtr> &outputs, void *stream_ptr) { in LaunchFirstHalf()
188 const std::vector<AddressPtr> &outputs, void *stream_ptr) { in LaunchSecondHalf()
|
| D | l2_loss_gpu_kernel.h | 37 const std::vector<AddressPtr> &outputs, void *stream_ptr) override { in Launch()
|
| D | gelu_kernel.h | 38 const std::vector<AddressPtr> &outputs, void *stream_ptr) override { in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | dynamic_shape_kernel.cc | 52 …micKernelPtr DynamicShapeKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) { in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | strided_slice_gpu_kernel.h | 40 const std::vector<AddressPtr> &outputs, void *stream_ptr) override { in Launch()
|
| D | strided_slice_grad_gpu_kernel.h | 40 const std::vector<AddressPtr> &outputs, void *stream_ptr) override { in Launch()
|
| D | zeroslike_gpu_kernel.h | 37 const std::vector<AddressPtr> &outputs, void *stream_ptr) override { in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/nvtit/ |
| D | md_iteration_gradient_descent_kernel.h | 52 const std::vector<AddressPtr> &outputs, void *stream_ptr) override { in Launch()
|
| D | md_iteration_setup_random_state.h | 53 const std::vector<AddressPtr> &outputs, void *stream_ptr) override { in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_mod.cc | 137 const std::vector<AddressPtr> &outputs, void *stream_ptr) { in Launch()
206 …:DynamicKernelPtr AicpuOpKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) { in GenDynamicKernel()
|