Searched refs:__cuda_array_interface__ (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/pytorch/test/ |
H A D | test_numba_integration.py | 67 self.assertRaises(AttributeError, lambda: cput.__cuda_array_interface__) 76 AttributeError, lambda: sparse_t.__cuda_array_interface__ 83 AttributeError, lambda: sparse_cuda_t.__cuda_array_interface__ 91 ar_dict = cudat.__cuda_array_interface__ 283 numba_ary.__cuda_array_interface__, 284 torch_ary.__cuda_array_interface__, 360 torch_ary.__cuda_array_interface__, numba_ary.__cuda_array_interface__ 382 torch_ary.__cuda_array_interface__, numba_ary.__cuda_array_interface__ 390 if1 = torch_ary.__cuda_array_interface__ 391 if2 = numba_ary.__cuda_array_interface__
|
/aosp_15_r20/external/pytorch/torch/ |
H A D | _tensor.py | 1185 def __cuda_array_interface__(self): member in Tensor 1194 Tensor.__cuda_array_interface__.__get__, # type: ignore[attr-defined]
|
H A D | overrides.py | 1379 Tensor.__cuda_array_interface__.__get__: lambda self: -1,
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/ |
H A D | __init__.pyi | 451 __cuda_array_interface__: Dict[str, Any]
|