Home
last modified time | relevance | path

Searched refs:usage_stream (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_device_buffer.cc191 se::Stream* usage_stream, std::shared_ptr<BufferSequencingEvent> event, in AddUsageEvent() argument
196 if (existing.stream == usage_stream) { in AddUsageEvent()
204 usage_events_.push_back({usage_stream, event, reference_held}); in AddUsageEvent()
Dtracked_device_buffer.h195 void AddUsageEvent(se::Stream* usage_stream,
Dpjrt_stream_executor_client.cc332 se::Stream* usage_stream, bool prefer_to_retain_reference) { in RecordUsage() argument
345 buffer_local_device->ThenRelease(usage_stream, device_buffer.buffer()); in RecordUsage()
347 device_buffer.ConvertUsageHold(usage_stream, event, in RecordUsage()
547 se::Stream* usage_stream, std::shared_ptr<BufferSequencingEvent> event, in ConvertUsageHold() argument
551 parent_->ConvertUsageHold(buffer().get(), usage_stream, std::move(event), in ConvertUsageHold()
1167 TrackedDeviceBuffer* buffer, se::Stream* usage_stream, in ConvertUsageHold() argument
1171 buffer->AddUsageEvent(usage_stream, std::move(event), reference_held); in ConvertUsageHold()
Dpjrt_stream_executor_client.h396 void ConvertUsageHold(se::Stream* usage_stream,
567 void ConvertUsageHold(TrackedDeviceBuffer* buffer, se::Stream* usage_stream,