- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for EnsureStreamOkLocked (0.13 sec)
-
tensorflow/compiler/jit/xla_device.cc
backend->stream_executors()[device_ordinal_]->CreateStream()); TF_RETURN_IF_ERROR(EnsureStreamOkLocked(backend, "stream", &stream_, &need_new_device_context)); (*global_compute_streams_)[device_ordinal_] = stream_; } } else { TF_RETURN_IF_ERROR(EnsureStreamOkLocked(backend, "stream", &stream_, &need_new_device_context));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.h
private: absl::StatusOr<xla::LocalClient*> GetOrCreateClient() const; Allocator* GetAllocatorLocked(AllocatorAttributes attr) TF_EXCLUSIVE_LOCKS_REQUIRED(mu_); Status EnsureStreamOkLocked(xla::Backend* backend, const string& name, std::shared_ptr<se::Stream>* stream, bool* stream_was_changed) TF_EXCLUSIVE_LOCKS_REQUIRED(mu_);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 13.4K bytes - Viewed (0)