Searched defs:StrideType (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_prims/ |
| H A D | __init__.py | 227 strides: Optional[StrideType] = None,
1216 a: TensorLikeType, size: ShapeType, stride: StrideType, storage_offset: int
1236 a: Tensor, size: ShapeType, stride: StrideType, storage_offset: int
1529 strides: Optional[StrideType] = None,
1592 strides: Optional[StrideType] = None,
1877 stride: StrideType,
2571 strides: StrideType,
|
| /aosp_15_r20/external/pytorch/torch/_prims_common/ |
| H A D | __init__.py | 38 StrideType: TypeAlias = Union[List[int], Tuple[int, ...]] variable
599 def validate_strides(strides: StrideType):
1737 shape: ShapeType, strides: StrideType, storage_offset: int
1778 a: torch.TypedStorage, shape: ShapeType, strides: StrideType, storage_offset: int
|
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | __init__.py | 2670 stride: StrideType,
2685 stride: StrideType,
3567 a_shape: ShapeType, a_stride: StrideType, dimension: int, size: int, step: int
4730 stride: StrideType,
5356 strides: StrideType,
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | debug_utils.py | 479 stride: Optional["torch._prims_common.StrideType"],
|
| /aosp_15_r20/external/deqp/external/vulkancts/modules/vulkan/query_pool/ |
| H A D | vktQueryPoolStatisticsTests.cpp | 94 enum StrideType enum
|