Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc558 CHECK_EQ(type_, kDonation); in ConfirmDonation()
569 if (type_ == kDonation) { in AddToInput()
1034 return holds_[ScopedHold::kDonation] == 0; in WaitForOutstandingDonationHold()
1129 if (type == ScopedHold::kDonation) { in GetBufferForHoldLocked()
1182 CHECK_EQ(holds_[ScopedHold::kDonation], 1); in ConfirmDonation()
1183 holds_[ScopedHold::kDonation] = 0; in ConfirmDonation()
1204 if (type == ScopedHold::kDonation) { in DropHold()
1205 CHECK_EQ(holds_[ScopedHold::kDonation], 0); in DropHold()
1278 if (type == ScopedHold::kDonation) { in GetBufferWithHold()
1285 if (type == ScopedHold::kDonation && !hold.ok()) { in GetBufferWithHold()
[all …]
Dpjrt_stream_executor_client.h333 enum Type { kUsage = 0, kExternalReference, kDonation, kMaxValue }; enumerator