Home
last modified time | relevance | path

Searched defs:dw_array (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dprelu_grad_impl.cu23 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/
Dprelu_grad_gpu_kernel.h49 auto *dw_array = GetDeviceAddress<float>(workspace, 0); in Launch() local