Searched refs:GetBufferForHoldLocked (Results 1 – 2 of 2) sorted by relevance
556 StatusOr<std::shared_ptr<TrackedDeviceBuffer>> GetBufferForHoldLocked(
1128 PjRtStreamExecutorBuffer::GetBufferForHoldLocked(ScopedHold::Type type) { in GetBufferForHoldLocked() function in xla::PjRtStreamExecutorBuffer1163 hold->Acquire(GetBufferForHoldLocked(hold->type())); in AcquireHoldLocked()