Searched refs:CudaArrayInterface (Results 1 – 2 of 2) sorted by relevance
125 StatusOr<pybind11::dict> CudaArrayInterface();
318 StatusOr<py::dict> PyBuffer::CudaArrayInterface() { in CudaArrayInterface() function in xla::PyBuffer674 [](PyBuffer::object self) { return self.buf()->CudaArrayInterface(); }); in RegisterTypes()