Searched defs:dbeta (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | group_norm_kernel.cu | 156 T* dbeta) { in GammaBeta1dBackwardCUDAKernel1()
194 T* dbeta) { in GammaBeta1dBackwardCUDAKernel2()
366 T* dbeta) { in GammaBetaBackwardCUDAKernel1()
402 T* dbeta) { in GammaBetaBackwardCUDAKernel2()
681 Tensor& dbeta) { in GroupNorm1dBackward()
808 Tensor& dbeta) { in GroupNormBackwardKernelImplInternal()
979 Tensor& dbeta) { in GroupNormBackwardKernelImpl()
|
| H A D | layer_norm_kernel.cu | 1143 Tensor* dbeta) { in LayerNormBackwardKernelImplInternal()
1320 Tensor* dbeta) { in LayerNormBackwardKernelImpl()
1413 Tensor dbeta; in layer_norm_backward_cuda() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | layer_norm_kernel.cpp | 227 [](Vec dbeta, Vec dy) { return dbeta + dy; }, in layer_norm_backward_frame()
343 [](fVec dbeta, fVec dy) { return dbeta + dy; }, in layer_norm_backward_frame()
501 Tensor* dbeta) { in LayerNormBackwardKernelImplInternal()
591 Tensor* dbeta) { in LayerNormBackwardKernelImpl()
|
| H A D | group_norm_kernel.cpp | 821 BetaBackward(int64_t N, int64_t C, const opmath_t* db, PT* dbeta) { in BetaBackward()
845 BetaBackward(int64_t N, int64_t C, const opmath_t* db, PT* dbeta) { in BetaBackward()
894 Tensor& dbeta) { in GroupNormBackwardKernelImplInternal()
1372 Tensor& dbeta) { in GroupNormBackwardKernelImplChannelsLastInternal()
1542 Tensor& dbeta) { in GroupNormBackwardKernelImpl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | group_norm.cpp | 132 Tensor dbeta; in native_group_norm_backward() local
|
| H A D | layer_norm.cpp | 138 Tensor dbeta; in layer_norm_backward_cpu() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorBackward.cpp | 221 Tensor dbeta; in layer_norm_backward_nested() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 4105 double dbeta = 0.0; in operator ()() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 4952 Tensor dbeta; in infinitely_differentiable_native_group_norm_backward() local
|