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]
|