Searched defs:DimsType (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/_refs/ |
| D | fft.py | 276 input: TensorLikeType, shape: Optional[ShapeType], dim: Optional[DimsType]
363 dim: Optional[DimsType] = None,
376 dim: Optional[DimsType] = None,
389 dim: Optional[DimsType] = None,
408 dim: Optional[DimsType] = None,
441 dim: Optional[DimsType],
470 dim: Optional[DimsType] = None,
487 dim: Optional[DimsType] = None,
508 dim: Optional[DimsType] = (-2, -1),
519 dim: Optional[DimsType] = (-2, -1),
[all …]
|
| D | __init__.py | 801 self: TensorLikeType, dim: DimsType, keepdim: bool = False
2146 dims: Optional[DimsType] = None,
2233 dim: Optional[DimsType] = None,
2248 dim: Optional[DimsType] = None,
2349 dim: Optional[DimsType] = None,
2373 dim: Optional[DimsType] = None,
2408 dim: Optional[DimsType] = None,
2459 dim: Optional[DimsType] = None,
2511 dim: Optional[DimsType] = None,
2537 dim: Optional[DimsType] = None,
[all …]
|
| /external/pytorch/torch/_refs/linalg/ |
| D | __init__.py | 105 dim: Optional[DimsType] = None,
177 dim: DimsType = (-2, -1),
259 dim: Optional[DimsType] = None,
|
| /external/pytorch/torch/_prims_common/ |
| D | __init__.py | 39 DimsType: TypeAlias = Union[int, List[int], Tuple[int, ...]] variable
|