Searched defs:DefaultPtrTraits (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cuh | 357 const GenericPackedTensorAccessor<const input_scalar_t, 3, DefaultPtrTraits, index_t> input, in batch_norm_backward_kernel() argument
474 const GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> input, in batch_norm_backward_reduce_kernel() argument
509 const GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> input, in batch_norm_backward_elemt_kernel_impl() argument
547 const GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> input, in batch_norm_backward_elemt_kernel() argument
569 const GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> input, in batch_norm_backward_elemt_kernel() argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | StridedRandomAccessor.h | 13 struct DefaultPtrTraits { struct
31 template <typename U> class PtrTraits = DefaultPtrTraits argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | TensorAccessor.h | 18 struct DefaultPtrTraits { struct
34 template<typename T, size_t N, template <typename U> class PtrTraits = DefaultPtrTraits, typename i… argument
|