Home
last modified time | relevance | path

Searched refs:CopyToHostAsync (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_buffer.h80 Status CopyToHostAsync();
Dpy_buffer.cc91 Status PyBuffer::CopyToHostAsync() { in CopyToHostAsync() function in xla::PyBuffer
141 TF_RETURN_IF_ERROR(CopyToHostAsync()); in AsNumPyArray()
Dxla.cc317 .def("copy_to_host_async", &PyBuffer::CopyToHostAsync) in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client_extension.cc159 .def("copy_to_host_async", &PyTpuBuffer::CopyToHostAsync, in PYBIND11_MODULE()
Dtpu_client.h222 Status CopyToHostAsync();
Dtpu_client.cc282 Status PyTpuBuffer::CopyToHostAsync() { in CopyToHostAsync() function in xla::PyTpuBuffer
346 TF_RETURN_IF_ERROR(CopyToHostAsync()); in ToLiteral()