Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dgrpc_tpu_driver.cc268 absl::MutexLock m(&request_lock_); in AddWriteRequest()
284 absl::Mutex request_lock_; member in tpu_driver::__anond0091eda0111::GrpcTpuStream
286 ABSL_GUARDED_BY(request_lock_);
287 int64_t num_pending_requests_ ABSL_GUARDED_BY(request_lock_) = 0;
289 bool shutting_down_ ABSL_GUARDED_BY(request_lock_) = false;
469 absl::MutexLock lock(&request_lock_); in ~GrpcTpuStream()
629 request_lock_.LockWhenWithTimeout( in StreamWriterFn()
633 request_lock_.Unlock(); in StreamWriterFn()
638 request_lock_.Unlock(); in StreamWriterFn()
657 request_lock_.Unlock(); in StreamWriterFn()