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