Searched defs:stat_scalar_t (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cuh | 217 const GenericPackedTensorAccessor<const stat_scalar_t, 1, RestrictPtrTraits, index_t> weight, in batch_norm_transform_input_kernel()
360 GenericPackedTensorAccessor<stat_scalar_t, 1, DefaultPtrTraits, index_t> grad_weight, in batch_norm_backward_kernel()
362 const GenericPackedTensorAccessor<const stat_scalar_t, 1, DefaultPtrTraits, index_t> weight, in batch_norm_backward_kernel()
480 GenericPackedTensorAccessor<stat_scalar_t, 1, DefaultPtrTraits, index_t> grad_weight, in batch_norm_backward_reduce_kernel()
513 const GenericPackedTensorAccessor<stat_scalar_t, 1, DefaultPtrTraits, index_t> weight, in batch_norm_backward_elemt_kernel_impl()
551 const GenericPackedTensorAccessor<stat_scalar_t, 1, DefaultPtrTraits, index_t> weight, in batch_norm_backward_elemt_kernel()
573 const GenericPackedTensorAccessor<stat_scalar_t, 1, DefaultPtrTraits, index_t> weight, in batch_norm_backward_elemt_kernel()
|