Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dsemaphore.h35 class ScopedReservation {
37 ScopedReservation(Semaphore* semaphore, int64 amount) in ScopedReservation() function
39 ~ScopedReservation();
41 ScopedReservation(const ScopedReservation&) = delete;
42 ScopedReservation(ScopedReservation&& other) noexcept;
43 ScopedReservation& operator=(const ScopedReservation&) = delete;
44 ScopedReservation& operator=(ScopedReservation&& other) noexcept;
52 ScopedReservation ScopedAcquire(int64 amount);
Dsemaphore.cc48 Semaphore::ScopedReservation::~ScopedReservation() { in ~ScopedReservation()
54 Semaphore::ScopedReservation::ScopedReservation( in ScopedReservation() function in xla::Semaphore::ScopedReservation
55 ScopedReservation&& other) noexcept { in ScopedReservation()
61 Semaphore::ScopedReservation& Semaphore::ScopedReservation::operator=( in operator =()
62 ScopedReservation&& other) noexcept { in operator =()
69 Semaphore::ScopedReservation Semaphore::ScopedAcquire(int64 amount) { in ScopedAcquire()
71 return ScopedReservation(this, amount); in ScopedAcquire()
Dpjrt_stream_executor_client.cc1768 auto compute_reservation = std::make_shared<Semaphore::ScopedReservation>( in EnqueueExecution()