Searched refs:RecordUsage (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 328 void RecordUsage(PjRtStreamExecutorBuffer::ScopedHold device_buffer, in RecordUsage() function
468 RecordUsage(py_buffer->GetBufferWithUsageHold(), local_device, local_device, in AllocateDestinationBuffer()
498 RecordUsage(std::move(device_buffer), local_device, local_device, in AddDestinationBufferSynchronization()
1262 RecordUsage(std::move(device_buffer), local_device, local_device, usage_event, in ToLiteral()
1421 RecordUsage(std::move(src_device_buffer), in CopyToDevice()
1559 RecordUsage(pjrt_buffer->GetBufferWithUsageHold(), local_device, local_device, in OutputBufferHelper()
1929 RecordUsage(std::move(b), device_state, device_state, definition_event, in ExecuteHelper()
|