Searched defs:stat_accscalar_t (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cuh | 215 …const GenericPackedTensorAccessor<typename std::conditional<train, stat_accscalar_t, stat_scalar_t… in batch_norm_transform_input_kernel() argument
273 GenericPackedTensorAccessor<stat_accscalar_t, 1, RestrictPtrTraits, index_t> save_mean, in batch_norm_collect_statistics_kernel()
365 … const GenericPackedTensorAccessor<const stat_accscalar_t, 1, DefaultPtrTraits, index_t> save_mean, in batch_norm_backward_kernel()
476 GenericPackedTensorAccessor<stat_accscalar_t, 1, DefaultPtrTraits, index_t> mean, in batch_norm_backward_reduce_kernel()
511 const GenericPackedTensorAccessor<stat_accscalar_t, 1, DefaultPtrTraits, index_t> mean, in batch_norm_backward_elemt_kernel_impl()
549 const GenericPackedTensorAccessor<stat_accscalar_t, 1, DefaultPtrTraits, index_t> mean, in batch_norm_backward_elemt_kernel()
571 const GenericPackedTensorAccessor<stat_accscalar_t, 1, DefaultPtrTraits, index_t> mean, in batch_norm_backward_elemt_kernel()
697 using stat_accscalar_t = at::acc_type<stat_scalar_t, true>; in batch_norm_elemt_cuda_template() typedef
782 using stat_accscalar_t = at::acc_type<stat_scalar_t, true>; in batch_norm_backward_reduce_cuda_template() typedef
842 using stat_accscalar_t = at::acc_type<stat_scalar_t, true>; in batch_norm_backward_elemt_cuda_template() typedef
[all …]
|