| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/opapi/aclnn/ |
| D | reduce_aclnn_kernel.cc | 31 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
40 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
63 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
|
| D | arg_with_value_aclnn_kernel.cc | 40 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
57 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
|
| D | batch_matmul_aclnn_kernel.cc | 35 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
51 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
|
| D | matmul_aclnn_kernel.cc | 40 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
57 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nccl/ |
| D | nccl_collective_gpu_kernel.h | 48 const std::vector<KernelTensor *> &outputs, void *stream_ptr) override { in Launch()
144 void *stream_ptr) { in LaunchAllReduce()
152 void *stream_ptr) { in LaunchAllGather()
160 void *stream_ptr) { in LaunchReduceScatter()
168 void *stream_ptr) { in LaunchBroadcast()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/convolution/ |
| D | depth_wise_gpu_kernel.h | 70 void *output_addr, void *stream_ptr) { in CallDepthWiseKernel()
182 … TransposeInfo info, void *output_addr, void *stream_ptr) { in CallTransposeToNHWC()
191 … TransposeInfo info, void *output_addr, void *stream_ptr) { in CallTransposeToNCHW()
200 … const std::vector<KernelTensor *> &workspace, void *stream_ptr) { in LaunchForward()
214 … const std::vector<KernelTensor *> &workspace, void *stream_ptr) { in LaunchInputGrad()
229 void *stream_ptr) { in LaunchFilterGrad()
|
| D | conv2d_cudnn_gpu_kernel.h | 611 void *padded_addr, void *stream_ptr) { in CalConv2dPadNCHW()
621 void *padded_addr, void *stream_ptr) { in CalConv2dPadNHWC()
631 void *dx_addr, void *stream_ptr) { in CalConv2dPadGradNCHW()
640 void *dx_addr, void *stream_ptr) { in CalConv2dPadGradNHWC()
651 … const std::vector<KernelTensor *> &workspace, void *stream_ptr) { in LaunchForward()
684 … const std::vector<KernelTensor *> &workspace, void *stream_ptr) { in LaunchInputGrad()
717 … const std::vector<KernelTensor *> &workspace, void *stream_ptr) { in LaunchFilterGrad()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/ |
| D | sort_gpu_kernel.cc | 27 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in LaunchKernel()
35 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in LaunchKernel()
111 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in LaunchKernel()
|
| D | scalegrad_gpu_kernel.cc | 23 … const std::vector<KernelTensor *> &outputs, void *stream_ptr, in LaunchScaleGradPerGrad()
42 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
|
| D | copy_with_slice_gpu_kernel.cc | 48 … const device::DeviceAddressPtr &dst_addr, void *stream_ptr) { in LaunchCopyWithSlice()
61 … const device::DeviceAddressPtr &dst_addr, void *stream_ptr) { in LaunchCopyWithSliceImpl()
|
| D | contiguous_gpu_kernel.cc | 53 … const device::DeviceAddressPtr &strides_addr, void *stream_ptr) { in LaunchContiguous()
69 void *stream_ptr) { in LaunchContiguousImpl()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/rl/ |
| D | priority_replay_buffer_gpu_kernel.cc | 77 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
112 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
147 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
193 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
220 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
|
| D | reservoir_replay_buffer_gpu_kernel.cc | 76 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
108 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
142 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
160 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | einsum_grad_gpu_kernel.h | 49 const std::vector<KernelTensor *> &outputs, void *stream_ptr) override { in Launch()
190 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in LaunchForward()
198 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in LaunchBackward()
204 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in ResOpBackward()
280 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in SingleOpBackward()
308 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in SingleOpForward()
343 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in ResOpForward()
|
| D | einsum_helper.h | 110 void *stream_ptr) { in SingleElementProcess()
121 const std::vector<size_t> &out_shape, void *stream_ptr) { in TwoElementProcess()
134 void *stream_ptr) { in SingleElementProcessGrad()
151 const std::vector<size_t> &dout_shape, void *stream_ptr) { in TwoElementProcessGrad()
168 … const std::vector<size_t> &rht_shape, const std::vector<size_t> &out_shape, void *stream_ptr) { in Mul()
200 const std::vector<size_t> &out_shape, void *stream_ptr) { in Bmm()
243 const std::vector<size_t> &out_shape, void *stream_ptr) { in Dot()
253 … const std::vector<size_t> &a_shape, const std::vector<size_t> &b_shape, void *stream_ptr) { in DotGrad()
265 const std::vector<size_t> &operate_info, void *stream_ptr) { in Diagonal()
283 const std::vector<size_t> &operate_info, void *stream_ptr) { in DiagonalGrad()
[all …]
|
| D | matrix_inverse_gpu_kernel.cc | 80 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in LaunchKernel_Cublas()
150 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in LaunchKernel_CuSolve()
233 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in LaunchKernel()
|
| D | multinomial_gpu_kernel.cc | 69 … const std::vector<kernel::KernelTensor *> &outputs, void *stream_ptr) { in Launch()
79 … const std::vector<kernel::KernelTensor *> &outputs, void *stream_ptr) { in LaunchKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | grid_sampler_gpu_kernel.h | 44 void *stream_ptr) { in LaunchKernel()
138 const std::vector<KernelTensor *> &outputs, void *stream_ptr) override { in Launch()
172 void *stream_ptr) { in LaunchKernel()
263 const std::vector<KernelTensor *> &outputs, void *stream_ptr) override { in Launch()
|
| D | binary_cross_entropy_grad_kernel.cc | 25 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
39 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in LaunchKernel()
|
| D | grid_sampler_grad_gpu_kernel.h | 42 void *stream_ptr) { in LaunchKernel()
64 const std::vector<KernelTensor *> &outputs, void *stream_ptr) override { in Launch()
204 void *stream_ptr) { in LaunchKernel()
225 const std::vector<KernelTensor *> &outputs, void *stream_ptr) override { in Launch()
|
| D | binary_cross_entropy_gpu_kernel.cc | 25 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
40 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in LaunchKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/ |
| D | hcom_gather.cc | 47 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
74 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in LaunchKernel()
|
| D | hcom_scatter.cc | 47 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
73 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in LaunchKernel()
|
| D | hcom_all_to_all.cc | 27 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
84 const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/quant/ |
| D | fake_quant_perchannel_gpu_kernel.cc | 97 float *scale, void *stream_ptr) { in CalFakeQuantize()
108 … const std::vector<KernelTensor *> &outputs, void *stream_ptr) { in Launch()
|