Searched defs:assume_unique (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorCompare.cpp | 454 bool assume_unique, in isin_sorting()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | generated_ops.cpp | 1001 const auto assume_unique = p_node->Input(2).toBool(); in __anon5d9c3eb96402() local
1019 const auto assume_unique = p_node->Input(2).toBool(); in __anon5d9c3eb96402() local
1037 const auto assume_unique = p_node->Input(2).toBool(); in __anon5d9c3eb96402() local
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 5016 def isin(elements, test_elements, *, assume_unique=False, invert=False): argument
5044 def isin_sorting(elements, test_elements, *, assume_unique=False, invert=False): argument
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 6442 def meta_isin(elements, test_elements, *, assume_unique=False, invert=False): argument
|