Searched defs:self_dim_stride (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | TensorCompareKernel.cpp | 80 auto self_dim_stride = ensure_nonempty_stride(self, dim); in compare_base_kernel() local
112 const scalar_t* self_data, auto self_dim_stride) { in min_kernel_impl()
145 const scalar_t* self_data, auto self_dim_stride) { in max_kernel_impl()
192 const scalar_t* self_data, auto self_dim_stride) { in aminmax_kernel()
245 auto self_dim_stride = ensure_nonempty_stride(self, dim); in mode_kernel_impl() local
|
| H A D | ReduceOpsKernel.cpp | 59 auto self_dim_stride = ensure_nonempty_stride(self, dim); in cpu_cum_base_kernel() local
86 const scalar_t* self_data, auto self_dim_stride, scalar_t init_val) { in cumsum_cpu_kernel()
105 const scalar_t* self_data, auto self_dim_stride, scalar_t init_val) { in cumprod_cpu_kernel()
124 const scalar_t* self_data, auto self_dim_stride, scalar_t init_val) { in logcumsumexp_cpu_kernel()
|
| H A D | ScatterGatherKernel.cpp | 192 auto self_dim_stride = ensure_nonempty_stride(buffer, dim); in operator ()() local
280 auto self_dim_stride = ensure_nonempty_stride(buffer, dim); in operator ()() local
376 auto self_dim_stride = ensure_nonempty_stride(buffer, dim); in operator ()() local
471 auto self_dim_stride = ensure_nonempty_stride(buffer, dim); in operator ()() local
567 auto self_dim_stride = ensure_nonempty_stride(buffer, dim); in operator ()() local
|
| H A D | IndexKernel.cpp | 198 int64_t self_dim_stride, in index_fill_kernel()
258 int64_t self_dim_stride) { in index_copy_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | IndexKernel.cu | 113 const int64_t self_dim_stride, in index_fill_kernel_impl()
151 const int64_t self_dim_stride) { in index_copy_kernel_impl()
207 const int64_t self_dim_stride, in index_fill_kernel()
223 const int64_t self_dim_stride) { in index_copy_kernel()
|
| H A D | ScatterGatherKernel.cu | 195 auto self_dim_stride = ensure_nonempty_stride(self, dim); in operator ()() local
253 auto self_dim_stride = ensure_nonempty_stride(self, dim); in operator ()() local
312 auto self_dim_stride = ensure_nonempty_stride(self, dim); in operator ()() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorAdvancedIndexing.cpp | 1545 auto self_dim_stride = (self_nonzero_dim.strides())[dim]; in index_fill_() local
|