Searched full:_unique2 (Results 1 – 17 of 17) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_meta.py | 747 torch.functional.unique: {f16}, # aten::_unique2, aten::unique_dim
842 aten._unique2.default : {i8, f64, i64, f16, bf16, f32, i32, b8, i16, u8, u16, u32, u64},
890 aten._unique2.default: {f16}, # aten::_unique2
|
| H A D | allowlist_for_publicAPI.json | 1314 "_unique2",
|
| H A D | test_autograd.py | 9023 res = torch._unique2(
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesDynamic.cpp | 72 UNSUPPORTED_DYNAMIC(_unique2); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/executorch/kernels/aten/ |
| H A D | functions.yaml | 31 - op: _unique2.out
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_has_decomposition.expect | 616 aten::_unique2
617 aten::_unique2.out
|
| /aosp_15_r20/external/pytorch/test/edge/ |
| H A D | selected_operators.yaml | 38 aten::_unique2.out:
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | Unique.mm | 11 #include <ATen/ops/_unique2.h>
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset11.py | 534 @_onnx_symbolic("aten::_unique2")
536 def _unique2(g: jit_utils.GraphContext, self, sorted, return_inverse, return_counts): function
|
| H A D | symbolic_opset9.py | 3311 @_onnx_symbolic("aten::_unique2")
3313 def _unique2(g: jit_utils.GraphContext, input, sorted, return_inverse, return_counts): function
3314 symbolic_helper._onnx_opset_unsupported("_unique2", 9, 11, input)
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | fake_impls.py | 334 @register_op_impl(aten._unique2.default)
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 45 - aten::_unique2
|
| /aosp_15_r20/frameworks/av/media/module/foundation/tests/ |
| H A D | AData_test.cpp | 437 std::unique_ptr<DerivedCounter> _unique, _unique2; in TEST_F() local
|
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | derivatives.yaml | 1835 - name: _unique2(Tensor self, bool sorted=True, bool return_inverse=False, bool return_counts=False…
1837 self: not_implemented("_unique2")
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | functional.py | 968 output, inverse_indices, counts = torch._unique2(
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | trace_rules.py | 1591 "torch._unique2",
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | native_functions.yaml | 6329 - func: _unique2(Tensor self, bool sorted=True, bool return_inverse=False, bool return_counts=False…
6336 autogen: _unique2.out
|