Searched refs:ConvertUsageHold (Results 1 – 2 of 2) sorted by relevance
396 void ConvertUsageHold(se::Stream* usage_stream,567 void ConvertUsageHold(TrackedDeviceBuffer* buffer, se::Stream* usage_stream,
347 device_buffer.ConvertUsageHold(usage_stream, event, in RecordUsage()546 void PjRtStreamExecutorBuffer::ScopedHold::ConvertUsageHold( in ConvertUsageHold() function in xla::PjRtStreamExecutorBuffer::ScopedHold551 parent_->ConvertUsageHold(buffer().get(), usage_stream, std::move(event), in ConvertUsageHold()1166 void PjRtStreamExecutorBuffer::ConvertUsageHold( in ConvertUsageHold() function in xla::PjRtStreamExecutorBuffer