Searched defs:DeviceLikeType (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | __init__.py | 1973 device: Optional[DeviceLikeType] = None,
4648 device: Optional[DeviceLikeType] = None,
4688 device: Optional[DeviceLikeType] = None,
4709 device: Optional[DeviceLikeType] = None,
4734 device: Optional[DeviceLikeType] = None,
4761 device: Optional[DeviceLikeType] = None,
4789 device: Optional[DeviceLikeType] = None,
4814 device: Optional[DeviceLikeType] = None,
4842 device: Optional[DeviceLikeType] = None,
4870 device: Optional[DeviceLikeType] = None,
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_prims_common/ |
| H A D | __init__.py | 57 DeviceLikeType: TypeAlias = Union[str, torch.device, int] variable
750 def canonicalize_device(device: DeviceLikeType) -> torch.device:
1919 def device_or_default(device: Optional[DeviceLikeType]) -> DeviceLikeType:
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | module.py | 1179 self: T, *, device: Optional[DeviceLikeType], recurse: bool = True
1199 device: Optional[DeviceLikeType] = ...,
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | _python_dispatch.py | 334 device: Optional["torch._prims_common.DeviceLikeType"] = None,
|