Searched defs:dw_array (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | prelu_grad_impl.cu | 23 const T *dy, const T *x, const T *w, T *dx, float *dw_array) { in CalPReLUGradKernel()
36 __global__ void InitDwArrayData(size_t dw_array_size, float *dw_array) { in InitDwArrayData()
43 __global__ void ComputeDwData(size_t weight_size, size_t thread_num, const float *dw_array, T *dw) { in ComputeDwData()
55 … const T *dy, const T *x, const T *w, T *dx, T *dw, float *dw_array, cudaStream_t cuda_stream) { in CalPReLUGrad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | prelu_grad_gpu_kernel.h | 49 auto *dw_array = GetDeviceAddress<float>(workspace, 0); in Launch() local
|