Searched defs:T_ACC (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | group_norm_kernel.cu | 38 using T_ACC = acc_type<T, true>; in RowwiseMomentsCUDAKernel() typedef
85 using T_ACC = acc_type<T, true>; in ComputeFusedParamsCUDAKernel() typedef
110 using T_ACC = acc_type<T, true>; in Compute1dBackwardFusedParamsCUDAKernel() typedef
157 using T_ACC = acc_type<T, true>; in GammaBeta1dBackwardCUDAKernel1() typedef
195 using T_ACC = acc_type<T, true>; in GammaBeta1dBackwardCUDAKernel2() typedef
284 using T_ACC = acc_type<T, true>; in ComputeInternalGradientsCUDAKernel() typedef
320 using T_ACC = acc_type<T, true>; in ComputeBackwardFusedParamsCUDAKernel() typedef
367 using T_ACC = acc_type<T, true>; in GammaBetaBackwardCUDAKernel1() typedef
403 using T_ACC = acc_type<T, true>; in GammaBetaBackwardCUDAKernel2() typedef
492 using T_ACC = acc_type<T, true>; in GroupNorm1dForward() typedef
[all …]
|
| D | BinaryMiscBackwardOpsKernels.cu | 63 using T_ACC = acc_type<scalar_t, true>; in logit_backward_kernel_cuda() typedef
|
| D | UnarySpecialOpsKernel.cu | 200 using T_ACC = acc_type<scalar_t, true>; in logit_kernel_cuda() typedef
|
| D | layer_norm_kernel.cu | 288 T_ACC /*eps*/, in vectorized_layer_norm_kernel_impl() argument
292 T_ACC* /*mean*/, in vectorized_layer_norm_kernel_impl() argument
1144 using T_ACC = acc_type<T, true>; in LayerNormBackwardKernelImplInternal() typedef
|
| /external/executorch/kernels/optimized/cpu/ |
| D | moments_utils.h | 93 using T_ACC = acc_t<T>; variable
|