Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc468 RecordUsage(py_buffer->GetBufferWithUsageHold(), local_device, local_device, in AllocateDestinationBuffer()
727 py_buffer->GetBufferWithUsageHold()); in BufferFromHostBuffer()
880 py_buffer->GetBufferWithUsageHold()); in BufferFromHostLiteral()
1307 ScopedHold dst_device_buffer(py_buffer->GetBufferWithUsageHold()); in CopyToDeviceHelper()
1559 RecordUsage(pjrt_buffer->GetBufferWithUsageHold(), local_device, local_device, in OutputBufferHelper()
Dpjrt_stream_executor_client.h511 ScopedHold GetBufferWithUsageHold() { in GetBufferWithUsageHold() function