Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.h333 enum Type { kUsage = 0, kExternalReference, kDonation, kMaxValue }; enumerator
515 return GetBufferWithHold(ScopedHold::kExternalReference); in GetBufferWithExternalReference()
Dpjrt_stream_executor_client.cc591 PjRtStreamExecutorBuffer::ScopedHold::kExternalReference); in ScopedHoldAsExternalReference()
1133 if (holds_[ScopedHold::kExternalReference] > 0) { in GetBufferForHoldLocked()
1181 CHECK_EQ(holds_[ScopedHold::kExternalReference], 0); in ConfirmDonation()
1207 CHECK_EQ(holds_[ScopedHold::kExternalReference], 0); in DropHold()