| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Unique.cu | 45 const bool return_inverse, in compute_unique()
93 const bool return_inverse, in unique_dim_cuda_template()
188 _unique_cuda(const Tensor& self, const bool sorted, const bool return_inverse) { in _unique_cuda()
198 _unique2_cuda(const Tensor& self, const bool sorted, const bool return_inverse, const bool return_c… in _unique2_cuda()
207 unique_dim_cuda(const Tensor& self, const int64_t dim, const bool sorted, const bool return_inverse… in unique_dim_cuda()
214 unique_dim_consecutive_cuda(const Tensor& self, const int64_t dim, const bool return_inverse, const… 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 | 67 const bool return_inverse, in compute_unique()
308 const bool return_inverse, in unique_cuda_template()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Unique.cpp | 37 const bool return_inverse, in unique_cpu_bool_template()
160 const bool return_inverse, in unique_cpu_sorted_template()
269 const bool return_inverse, in unique_consecutive_cpu_template()
361 const bool return_inverse, in _unique_dim_cpu_template()
442 _unique_cpu(const Tensor& self, const bool sorted, const bool return_inverse) { in _unique_cpu()
458 _unique2_cpu(const Tensor& self, const bool sorted, const bool return_inverse, const bool return_co… in _unique2_cpu()
471 unique_dim_cpu(const Tensor& self, const int64_t dim, const bool sorted, const bool return_inverse,… in unique_dim_cpu()
479 unique_dim_consecutive_cpu(const Tensor& self, const int64_t dim, const bool return_inverse, const … 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 | 1050 return_inverse=False, argument
1066 return_inverse=False, argument
1079 def _return_inverse( argument
1134 return_inverse=False, argument
1151 return_inverse=False, argument
1164 def _consecutive_return_inverse( 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 | 3307 def _unique(g: jit_utils.GraphContext, input, sorted, return_inverse): argument
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 | 951 return_inverse=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
|