Searched refs:GetOrCreateSubStream (Results 1 – 7 of 7) sorted by relevance
51 Stream* sub_stream1 = stream.GetOrCreateSubStream(); in TEST_F()56 Stream* sub_stream2 = stream.GetOrCreateSubStream(); in TEST_F()72 Stream* sub_stream1 = stream.GetOrCreateSubStream(); in TEST_F()74 Stream* sub_stream2 = stream.GetOrCreateSubStream(); in TEST_F()83 Stream* sub_stream3 = stream.GetOrCreateSubStream(); in TEST_F()90 Stream* sub_stream4 = stream.GetOrCreateSubStream(); in TEST_F()103 Stream* sub_stream1 = stream.GetOrCreateSubStream(); in TEST_F()113 Stream* sub_stream2 = stream.GetOrCreateSubStream(); in TEST_F()130 Stream* sub_stream3 = stream.GetOrCreateSubStream(); in TEST_F()142 Stream* sub_stream1 = stream.GetOrCreateSubStream(); in TEST_F()[all …]
158 Stream *GetOrCreateSubStream() TF_LOCKS_EXCLUDED(mu_);
922 Stream *Stream::GetOrCreateSubStream() { in GetOrCreateSubStream() function in stream_executor::Stream
55 se::Stream* substream = stream->GetOrCreateSubStream(); in TransferLiteralFromDevice()80 se::Stream* substream = stream->GetOrCreateSubStream(); in TransferLiteralFromDevice()103 se::Stream* substream = stream->GetOrCreateSubStream(); in TransferLiteralToDevice()118 se::Stream* substream = stream->GetOrCreateSubStream(); in TransferArrayFromDevice()145 se::Stream* substream = stream->GetOrCreateSubStream(); in TransferArrayToDevice()
376 se::Stream* deregister_stream = stream->GetOrCreateSubStream(); in UnregisterCancellation()
179 ? device_to_device_master_stream->GetOrCreateSubStream() in TpuDeviceToDeviceCopy()
331 auto stream2 = stream_->GetOrCreateSubStream(); in XLA_TEST_F()