Searched defs:DimsType (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A 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 …]
|
| H A 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 …]
|
| /aosp_15_r20/external/pytorch/torch/_refs/linalg/ |
| H A D | __init__.py | 105 dim: Optional[DimsType] = None,
177 dim: DimsType = (-2, -1),
259 dim: Optional[DimsType] = None,
|
| /aosp_15_r20/external/pytorch/torch/_prims_common/ |
| H A D | __init__.py | 39 DimsType: TypeAlias = Union[int, List[int], Tuple[int, ...]] variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/ |
| D | TestPartitioning.cpp | 2449 using DimsType = std::vector<uint32_t>; in executeCompilationAndCompareOutput() typedef
|