Searched defs:inverse_indices (Results 1 – 3 of 3) sorted by relevance
44 Tensor inverse_indices = at::empty({0}, self.options().dtype(kLong)); in unique_cpu_bool_template() local167 Tensor inverse_indices = at::empty({0}, self.options().dtype(kLong)); in unique_cpu_sorted_template() local275 Tensor inverse_indices = at::empty({0}, self.options().dtype(kLong)); in unique_consecutive_cpu_template() local374 Tensor inverse_indices = in _unique_dim_cpu_template() local421 Tensor inverse_indices = at::empty(indices.size(), self.options().dtype(kLong)); in _unique_dim_cpu_template() local
76 Tensor inverse_indices; in compute_unique() local247 Tensor output, inverse_indices, counts; in operator ()() local315 Tensor inverse_indices = at::empty(self.sizes(), self.options().dtype(kLong)); in unique_cuda_template() local
52 Tensor inverse_indices; in compute_unique() local120 Tensor inverse_indices = in unique_dim_cuda_template() local