Searched defs:dgamma (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | group_norm_kernel.cu | 155 T* dgamma, in GammaBeta1dBackwardCUDAKernel1()
193 T* dgamma, in GammaBeta1dBackwardCUDAKernel2()
365 T* dgamma, in GammaBetaBackwardCUDAKernel1()
401 T* dgamma, in GammaBetaBackwardCUDAKernel2()
680 Tensor& dgamma, in GroupNorm1dBackward()
807 Tensor& dgamma, in GroupNormBackwardKernelImplInternal()
978 Tensor& dgamma, in GroupNormBackwardKernelImpl()
|
| D | layer_norm_kernel.cu | 1142 Tensor* dgamma, in LayerNormBackwardKernelImplInternal()
1319 Tensor* dgamma, in LayerNormBackwardKernelImpl()
1412 Tensor dgamma; in layer_norm_backward_cuda() local
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | layer_norm_kernel.cpp | 212 [a, b](Vec dgamma, Vec dy, Vec x) { in layer_norm_backward_frame()
328 [a, b](fVec dgamma, fVec dy, fVec x) { in layer_norm_backward_frame()
500 Tensor* dgamma, in LayerNormBackwardKernelImplInternal()
590 Tensor* dgamma, in LayerNormBackwardKernelImpl()
|
| D | group_norm_kernel.cpp | 720 PT* dgamma) { in GammaBackward()
767 PT* dgamma) { in GammaBackward()
893 Tensor& dgamma, in GroupNormBackwardKernelImplInternal()
1371 Tensor& dgamma, in GroupNormBackwardKernelImplChannelsLastInternal()
1541 Tensor& dgamma, in GroupNormBackwardKernelImpl()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | group_norm.cpp | 131 Tensor dgamma; in native_group_norm_backward() local
|
| D | layer_norm.cpp | 137 Tensor dgamma; in layer_norm_backward_cpu() local
|
| /external/pytorch/aten/src/ATen/native/nested/ |
| D | NestedTensorBackward.cpp | 220 Tensor dgamma; in layer_norm_backward_nested() local
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 4951 Tensor dgamma; in infinitely_differentiable_native_group_norm_backward() local
|