Searched defs:__dlpack_device__ (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_dlpack.py | 179 def __dlpack_device__(self): member in TestTorchDlPack.test_dlpack_default_stream.DLPackTensor |
| /aosp_15_r20/external/pytorch/torch/_numpy/ | ||
| H A D | _ndarray.py | 489 def __dlpack_device__(self): member in ndarray |
| /aosp_15_r20/external/pytorch/torch/ | ||
| H A D | _tensor.py | 1578 def __dlpack_device__(self) -> Tuple[enum.IntEnum, int]: member in Tensor |