Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_buffer.cc92 if (!buffer_->IsOnCpu() && !host_value_) { in CopyToHostAsync()
94 host_value_ = host_value; in CopyToHostAsync()
142 if (!host_value_->ready.HasBeenNotified()) { in AsNumPyArray()
144 host_value_->ready.WaitForNotification(); in AsNumPyArray()
146 TF_RETURN_IF_ERROR(host_value_->status); in AsNumPyArray()
147 TF_ASSIGN_OR_RETURN(py::object array, LiteralToPython(host_value_->value)); in AsNumPyArray()
Dpy_buffer.h75 host_value_ = nullptr; in Delete()
123 std::shared_ptr<HostValue> host_value_; // Protected by the GIL. variable
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc279 host_value_ = nullptr; in Delete()
292 if (host_value_) { in CopyToHostAsync()
299 host_value_ = host_value; in CopyToHostAsync()
349 std::shared_ptr<HostValue> host_value = host_value_; in ToLiteral()
Dtpu_client.h284 std::shared_ptr<HostValue> host_value_ TF_GUARDED_BY(mu_);