Searched refs:BlockHostUntilReady (Results 1 – 10 of 10) sorted by relevance
85 Status PyBuffer::BlockHostUntilReady() { in BlockHostUntilReady() function in xla::PyBuffer88 return buffer_->BlockHostUntilReady(); in BlockHostUntilReady()136 TF_RETURN_IF_ERROR(buffer_->BlockHostUntilReady()); in AsNumPyArray()295 TF_RETURN_IF_ERROR(buffer.BlockHostUntilReady()); in PjRtBufferGetBuffer()
79 Status BlockHostUntilReady();
313 TF_RETURN_IF_ERROR(buffer->BlockHostUntilReady()); in PYBIND11_MODULE()316 .def("block_host_until_ready", &PyBuffer::BlockHostUntilReady) in PYBIND11_MODULE()
280 TF_RETURN_IF_ERROR(buffer->BlockHostUntilReady()); in BufferToDLPackManagedTensor()
385 virtual Status BlockHostUntilReady() = 0;
523 Status BlockHostUntilReady() override;
1436 Status PjRtStreamExecutorBuffer::BlockHostUntilReady() { in BlockHostUntilReady() function in xla::PjRtStreamExecutorBuffer
157 return buffer->BlockHostUntilReady(); in PYBIND11_MODULE()
242 Status BlockHostUntilReady();
428 Status PyTpuBuffer::BlockHostUntilReady() { in BlockHostUntilReady() function in xla::PyTpuBuffer