Searched defs:self_dim_size (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | IndexKernel.cu | 112 const int64_t self_dim_size, in index_fill_kernel_impl()
150 const int64_t self_dim_size, in index_copy_kernel_impl()
206 const int64_t self_dim_size, in index_fill_kernel()
222 const int64_t self_dim_size, in index_copy_kernel()
|
| H A D | ScatterGatherKernel.cu | 196 auto self_dim_size = ensure_nonempty_size(self, dim); in operator ()() local
254 auto self_dim_size = ensure_nonempty_size(self, dim); in operator ()() local
313 auto self_dim_size = ensure_nonempty_size(self, dim); in operator ()() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | TensorCompareKernel.cpp | 107 int64_t self_dim_size = ensure_nonempty_size(self, dim); in min_kernel_impl() local
140 int64_t self_dim_size = ensure_nonempty_size(self, dim); in max_kernel_impl() local
174 int64_t self_dim_size = ensure_nonempty_size(self, wrap_dim); in aminmax_kernel() local
244 auto self_dim_size = ensure_nonempty_size(self, dim); in mode_kernel_impl() local
|
| H A D | ScatterGatherKernel.cpp | 193 auto self_dim_size = ensure_nonempty_size(buffer, dim); in operator ()() local
281 auto self_dim_size = ensure_nonempty_size(buffer, dim); in operator ()() local
377 auto self_dim_size = ensure_nonempty_size(buffer, dim); in operator ()() local
472 auto self_dim_size = ensure_nonempty_size(buffer, dim); in operator ()() local
568 auto self_dim_size = ensure_nonempty_size(buffer, dim); in operator ()() local
|
| H A D | ReduceOpsKernel.cpp | 81 int64_t self_dim_size = ensure_nonempty_size(self, wrap_dim); in cumsum_cpu_kernel() local
100 int64_t self_dim_size = ensure_nonempty_size(self, wrap_dim); in cumprod_cpu_kernel() local
119 int64_t self_dim_size = ensure_nonempty_size(self, wrap_dim); in logcumsumexp_cpu_kernel() local
|
| H A D | IndexKernel.cpp | 197 int64_t self_dim_size, in index_fill_kernel()
257 int64_t self_dim_size, in index_copy_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorDimApply.h | 19 int self_dim_size = self.size(dim); in tensor_dim_apply3() local
|
| H A D | TensorAdvancedIndexing.cpp | 1018 auto self_dim_size = result.size(dim); in TORCH_IMPL_FUNC() local
1115 auto self_dim_size = result.size(dim); in index_reduce_func_impl() local
1352 auto self_dim_size = self.size(dim); in index_select_out_cpu_() local
1544 auto self_dim_size = (self_nonzero_dim.sizes())[dim]; in index_fill_() local
|
| H A D | ReduceOps.cpp | 773 int self_dim_size, int self_stride, int values_stride, int indices_stride) { in cummax_cummin_helper()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorMatmul.cpp | 271 const auto& self_dim_size = self_sizes.select(1, -1); in matmul_nested() local
|