- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for BorrowStream (0.25 sec)
-
tensorflow/compiler/jit/xla_device_context.cc
if (device_to_host_stream_) { device_to_host_stream = device_to_host_stream_; } else { absl::StatusOr<xla::StreamPool::Ptr> ptr_or_status = client_->mutable_backend()->BorrowStream( stream_->parent()->device_ordinal()); if (!ptr_or_status.status().ok()) { done(ptr_or_status.status()); return; } device_to_host_stream =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.cc
bool* stream_was_changed) { if (!(*stream) || !(*stream)->ok()) { xla::StreamPool::Ptr ptr; TF_ASSIGN_OR_RETURN(ptr, backend->BorrowStream(device_ordinal_)); *stream = std::shared_ptr<se::Stream>(std::move(ptr)); VLOG(1) << "XlaDevice " << this << " new " << name << " stream=" << (*stream); *stream_was_changed = true; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0)