Searched refs:TensorSequenceType (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | __init__.py | 41 TensorSequenceType,
2717 def cat(tensors: TensorSequenceType, dim: int = 0) -> TensorLikeType: argument
2837 def column_stack(tensors: TensorSequenceType) -> TensorLikeType: argument
2953 def dstack(tensors: TensorSequenceType) -> TensorLikeType: argument
3872 def _check_stack_inputs(tensors: TensorSequenceType) -> None: argument
3883 def stack(tensors: TensorSequenceType, dim: int = 0) -> TensorLikeType: argument
3920 def hstack(tensors: TensorSequenceType) -> TensorLikeType: argument
3930 def vstack(tensors: TensorSequenceType) -> TensorLikeType: argument
3944 def unbind(t: TensorLikeType, dim: int = 0) -> TensorSequenceType:
4445 def dsplit(a: TensorLikeType, sections: DimsType) -> TensorSequenceType:
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 26 TensorSequenceType,
3708 def _compute_weight_precision(weights: TensorSequenceType) -> Tensor: argument
3967 def _upsample_get_cubic_coefficients(t: Tensor) -> TensorSequenceType:
3987 def _upsample_cubic_interp1d(coeffs: TensorSequenceType, ts: Tensor) -> Tensor: argument
4157 def clip(xs: Tensor, ys: Tensor, ws: Tensor) -> TensorSequenceType:
|
| /aosp_15_r20/external/pytorch/torch/_prims_common/ |
| H A D | __init__.py | 93 TensorSequenceType: TypeAlias = Union[List[TensorLikeType], Tuple[TensorLikeType, ...]] variable
|