Searched defs:DimsSequenceType (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_prims/ |
| H A D | __init__.py | 1496 a: TensorLikeType, dimensions: DimsSequenceType, ndim=None
1527 start_indices: DimsSequenceType,
1528 limit_indices: DimsSequenceType,
1590 start_indices: DimsSequenceType,
1591 limit_indices: DimsSequenceType,
1788 def _transpose_meta(a: TensorLikeType, permutation: DimsSequenceType) -> TensorLikeType:
1806 def _transpose_aten(a: Tensor, permutation: DimsSequenceType) -> Tensor:
2029 def _rev_meta(a: TensorLikeType, dims: DimsSequenceType) -> TensorLikeType:
2426 dims: Optional[DimsSequenceType],
2442 dims: Optional[DimsSequenceType],
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_prims_common/ |
| H A D | __init__.py | 40 DimsSequenceType: TypeAlias = Union[List[int], Tuple[int, ...]] variable
621 def validate_dimension_indices(rank: int, indices: DimsSequenceType):
689 def is_valid_permutation(rank: int, perm: DimsSequenceType) -> bool:
819 dims: Union[DimsSequenceType, Tuple[DimsSequenceType, ...]]
|
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | __init__.py | 3045 def flip(a: TensorLikeType, dims: DimsSequenceType) -> TensorLikeType:
3843 a: TensorLikeType, k: int = 1, dims: DimsSequenceType = (0, 1)
5291 source: Union[int, DimsSequenceType], argument
|