Searched defs:RestrictPtrTraits (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cuh | 213 const GenericPackedTensorAccessor<const input_scalar_t, 3, RestrictPtrTraits, index_t> input, in batch_norm_transform_input_kernel() argument
270 const GenericPackedTensorAccessor<const input_scalar_t, 3, RestrictPtrTraits, index_t> input, in batch_norm_collect_statistics_kernel() argument
428 const GenericPackedTensorAccessor<accscalar_t, 2, RestrictPtrTraits, index_t> vec_mean, in batch_norm_reduce_statistics_kernel() argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | StridedRandomAccessor.h | 24 struct RestrictPtrTraits { struct
25 using PtrType = T* RESTRICT;
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | TensorAccessor.h | 24 struct RestrictPtrTraits { struct
25 typedef T* __restrict__ PtrType;
|