Home
last modified time | relevance | path

Searched refs:__cuda_array_interface__ (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_numba_integration.py67 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.py1185 def __cuda_array_interface__(self): member in Tensor
1194 Tensor.__cuda_array_interface__.__get__, # type: ignore[attr-defined]
H A Doverrides.py1379 Tensor.__cuda_array_interface__.__get__: lambda self: -1,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/
H A D__init__.pyi451 __cuda_array_interface__: Dict[str, Any]