| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Unique.cu | 46 const bool return_counts, in compute_unique()
94 const bool return_counts in unique_dim_cuda_template()
198 …_cuda(const Tensor& self, const bool sorted, const bool return_inverse, const bool return_counts) { in _unique2_cuda()
207 … self, const int64_t dim, const bool sorted, const bool return_inverse, const bool return_counts) { in unique_dim_cuda()
214 …_cuda(const Tensor& self, const int64_t dim, const bool return_inverse, const bool return_counts) { in unique_dim_consecutive_cuda()
221 unique_consecutive_cuda(const Tensor& self, const bool return_inverse, const bool return_counts, st… in unique_consecutive_cuda()
|
| H A D | UniqueCub.cu | 68 const bool return_counts, in compute_unique()
309 const bool return_counts) { in unique_cuda_template()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Unique.cpp | 38 const bool return_counts) { in unique_cpu_bool_template()
161 const bool return_counts, in unique_cpu_sorted_template()
270 const bool return_counts) { in unique_consecutive_cpu_template()
362 const bool return_counts) { in _unique_dim_cpu_template()
458 …2_cpu(const Tensor& self, const bool sorted, const bool return_inverse, const bool return_counts) { in _unique2_cpu()
471 … self, const int64_t dim, const bool sorted, const bool return_inverse, const bool return_counts) { in unique_dim_cpu()
479 …e_cpu(const Tensor& self, const int64_t dim, const bool return_inverse, const bool return_counts) { in unique_dim_consecutive_cpu()
486 unique_consecutive_cpu(const Tensor& self, const bool return_inverse, const bool return_counts, std… in unique_consecutive_cpu()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | functional.py | 1047 def _return_counts( argument
1067 return_counts=False, argument
1083 return_counts=False, argument
1132 def _consecutive_return_counts( argument
1152 return_counts=False, argument
1167 return_counts=False, argument
|
| H A D | _tensor.py | 985 def unique(self, sorted=True, return_inverse=False, return_counts=False, dim=None): argument
1008 def unique_consecutive(self, return_inverse=False, return_counts=False, dim=None): argument
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | fake_impls.py | 272 fake_mode, func, arg, dim, sorted=True, return_inverse=False, return_counts=False argument
336 fake_mode, func, arg, sorted=True, return_inverse=False, return_counts=False argument
343 fake_mode, func, arg, dim, sorted=True, return_inverse=False, return_counts=False argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset11.py | 536 def _unique2(g: jit_utils.GraphContext, self, sorted, return_inverse, return_counts): argument
546 g: jit_utils.GraphContext, self, dim, sorted, return_inverse, return_counts argument
|
| H A D | symbolic_opset9.py | 3313 def _unique2(g: jit_utils.GraphContext, input, sorted, return_inverse, return_counts): argument
|
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _funcs_impl.py | 952 return_counts=False, argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 4198 def helper(x, return_inverse, return_counts): argument
4218 def helper(x, dim, return_inverse, return_counts): argument
|