Searched defs:__dlpack__ (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_dlpack.py | 182 def __dlpack__(self, stream=None): member in TestTorchDlPack.test_dlpack_default_stream.DLPackTensor |
| /aosp_15_r20/external/pytorch/torch/_numpy/ | ||
| H A D | _ndarray.py | 486 def __dlpack__(self, *, stream=None): member in ndarray |
| /aosp_15_r20/external/pytorch/torch/ | ||
| H A D | _tensor.py | 1520 def __dlpack__(self, stream=None): member in Tensor |