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