Searched refs:GetBufferWithHold (Results 1 – 2 of 2) sorted by relevance
510 ScopedHold GetBufferWithHold(ScopedHold::Type type);512 return GetBufferWithHold(ScopedHold::kUsage); in GetBufferWithUsageHold()515 return GetBufferWithHold(ScopedHold::kExternalReference); in GetBufferWithExternalReference()
1277 PjRtStreamExecutorBuffer::GetBufferWithHold(ScopedHold::Type type) { in GetBufferWithHold() function in xla::PjRtStreamExecutorBuffer1709 device_buffers->emplace_back(handle->GetBufferWithHold( in EnqueueExecution()