Home
last modified time | relevance | path

Searched defs:usage_stream (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtracked_device_buffer.cc193 se::Stream* usage_stream, std::shared_ptr<BufferSequencingEvent> event, in AddUsageEvent()
H A Dpjrt_stream_executor_client.cc335 se::Stream* usage_stream, bool prefer_to_retain_reference, in RecordUsage()
557 se::Stream* usage_stream, std::shared_ptr<BufferSequencingEvent> event, in ConvertUsageHold()
1270 TrackedDeviceBuffer* buffer, se::Stream* usage_stream, in ConvertUsageHold()