| /aosp_15_r20/external/pytorch/torch/ |
| H A D | functional.py | 42 "unique_consecutive",
819 .. note:: This function is different from :func:`torch.unique_consecutive` in the sense that
825 :func:`torch.unique_consecutive` which avoids the sorting.
1016 >>> output = torch.unique_consecutive(x)
1020 >>> output, inverse_indices = torch.unique_consecutive(x, return_inverse=True)
1026 >>> output, counts = torch.unique_consecutive(x, return_counts=True)
1034 unique_consecutive,
1041 output, inverse_indices, counts = _VF.unique_consecutive( # type: ignore[attr-defined]
1188 func_name="unique_consecutive",
1198 func_name="unique_consecutive",
[all …]
|
| H A D | _tensor.py | 1008 def unique_consecutive(self, return_inverse=False, return_counts=False, dim=None): member in Tensor
1011 See :func:`torch.unique_consecutive`
1015 Tensor.unique_consecutive,
1022 return torch.unique_consecutive(
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _builtins.py | 111 (torch._VF.unique_consecutive, "aten::unique_consecutive"), # type: ignore[attr-defined]
134 "unique_consecutive",
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_meta.py | 674 …torch.functional.unique_consecutive : {f64, i32, i64, u8, i16, f16, bf16, b8, i8, f32, u16, u32, u…
748 torch.functional.unique_consecutive: {f16}, # aten::unique_consecutive
847 … aten.unique_consecutive.default : {i8, f64, i64, f16, bf16, f32, i32, b8, i16, u8, u16, u32, u64},
898 aten.unique_consecutive.default: {f16}, # aten::unique_consecutive
|
| H A D | test_sort_and_select.py | 755 y_unique, y_inverse, y_counts = torch.unique_consecutive(
767 y_unique, y_inverse, y_counts = torch.unique_consecutive(
1056 torch.unique_consecutive,
1057 lambda x, **kwargs: x.unique_consecutive(**kwargs),
|
| H A D | test_proxy_tensor.py | 2007 …xfail('unique_consecutive', ''), # aten.unique_consecutive.default - couldn't find symbolic meta …
|
| H A D | test_ops.py | 2351 "unique_consecutive",
|
| /aosp_15_r20/external/executorch/docs/source/ |
| H A D | concepts.md | 120 …rence. For example, the ATen op [unique_consecutive](https://pytorch.org/docs/stable/generated/tor…
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesDynamic.cpp | 70 UNSUPPORTED_DYNAMIC(unique_consecutive); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | public_api | 539 unique_consecutive
|
| H A D | annotated_ops | 292 unique_consecutive, misc
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 614 - aten::unique_consecutive
997 aten::unique_consecutive: 2
|
| H A D | model_ops.yaml | 374 aten::unique_consecutive: 2
|
| H A D | math_ops.py | 254 torch.unique_consecutive(a),
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_has_decomposition.expect | 1310 aten::unique_consecutive
1311 aten::unique_consecutive.out
|
| /aosp_15_r20/external/executorch/examples/qualcomm/scripts/ |
| H A D | wav2letter.py | 90 torch.unique_consecutive(pred[i, :, :].argmax(0))
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | torch.rst | 453 unique_consecutive
|
| H A D | tensors.rst | 770 Tensor.unique_consecutive
|
| H A D | conf.py | 663 "unique_consecutive",
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | Unique.mm | 14 #include <ATen/ops/unique_consecutive.h>
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | top_operators_github_usage.py | 217 ("unique_consecutive", 471),
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_dtensor_ops.py | 460 xfail("unique_consecutive"),
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LinearAlgebra.cpp | 141 #include <ATen/ops/unique_consecutive.h>
2587 // Then we call `unique_consecutive` and we will use it to split `sorted_s`, in compute_T18_scale_square()
2589 auto split_counts = std::get<2>(at::unique_consecutive(sorted_s, true, /*return_counts=*/true)); in compute_T18_scale_square()
|
| /aosp_15_r20/external/pytorch/tools/pyi/ |
| H A D | gen_pyi.py | 152 "unique_consecutive",
|
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | derivatives.yaml | 1827 - name: unique_consecutive(Tensor self, bool return_inverse=False, bool return_counts=False, int? d…
1829 self: not_implemented("unique_consecutive")
|