Searched refs:ScopedHold (Results 1 – 2 of 2) sorted by relevance
331 class ScopedHold {347 ~ScopedHold();348 ScopedHold(ScopedHold&& other);349 ScopedHold(const ScopedHold&) = delete;350 ScopedHold& operator=(const ScopedHold&) = delete;426 ScopedHold(PjRtStreamExecutorBuffer* parent, Type type) in ScopedHold() function428 explicit ScopedHold(const ForClosure& closure_helper) in ScopedHold() function510 ScopedHold GetBufferWithHold(ScopedHold::Type type);511 ScopedHold GetBufferWithUsageHold() { in GetBufferWithUsageHold()512 return GetBufferWithHold(ScopedHold::kUsage); in GetBufferWithUsageHold()[all …]
328 void RecordUsage(PjRtStreamExecutorBuffer::ScopedHold device_buffer, in RecordUsage()481 PjRtStreamExecutorBuffer::ScopedHold device_buffer, in AddDestinationBufferSynchronization()506 PjRtStreamExecutorBuffer::ScopedHold::~ScopedHold() { in ~ScopedHold()512 PjRtStreamExecutorBuffer::ScopedHold::ScopedHold(ScopedHold&& other) in ScopedHold() function in xla::PjRtStreamExecutorBuffer::ScopedHold522 void PjRtStreamExecutorBuffer::ScopedHold::Acquire( in Acquire()537 PjRtStreamExecutorBuffer::ScopedHold::ForClosure538 PjRtStreamExecutorBuffer::ScopedHold::ToClosure() { in ToClosure()546 void PjRtStreamExecutorBuffer::ScopedHold::ConvertUsageHold( in ConvertUsageHold()556 void PjRtStreamExecutorBuffer::ScopedHold::ConfirmDonation() { in ConfirmDonation()563 void PjRtStreamExecutorBuffer::ScopedHold::AddToInput( in AddToInput()[all …]