Searched defs:self_i (Results 1 – 4 of 4) sorted by relevance
1024 auto self_i = index_data[i]; in TORCH_IMPL_FUNC() local1051 auto self_i = index_data[i]; in TORCH_IMPL_FUNC() local1121 auto self_i = index_data[i]; in index_reduce_func_impl() local1175 auto self_i = index_data[i]; in index_reduce_func_impl() local1374 auto self_i = index_data[i]; in index_select_out_cpu_() local1403 auto self_i = index_data[i]; in index_select_out_cpu_() local1430 auto self_i = index_data[i]; in index_select_out_cpu_() local1450 auto self_i = index_data[i]; in index_select_out_cpu_() local
263 for (size_t self_i = pos; self_i < self.size(); ++self_i) { in find_first_not_of() local331 for (size_t self_i = std::min(pos, self.size() - 1); ; in find_last_of() local376 for (size_t self_i = std::min(pos, self.size() - 1); ; --self_i) { in find_last_not_of() local
169 const scalar_t * self_i = self_start + i * m; in run_parallel_pdist() local229 const scalar_t * self_i = t1_start + size1 * l + i; in run_parallel_cdist() local
143 const scalar_t * self_i = start + init; in cdist_backward_kernel_cuda_impl() local176 const scalar_t * self_i = start + init; in pdist_backward_kernel_cuda_impl() local