Searched refs:GetOrCreateSubStream (Results 1 – 7 of 7) sorted by relevance
53 Stream* sub_stream1 = stream.GetOrCreateSubStream(); in TEST_F()58 Stream* sub_stream2 = stream.GetOrCreateSubStream(); in TEST_F()74 Stream* sub_stream1 = stream.GetOrCreateSubStream(); in TEST_F()76 Stream* sub_stream2 = stream.GetOrCreateSubStream(); in TEST_F()85 Stream* sub_stream3 = stream.GetOrCreateSubStream(); in TEST_F()92 Stream* sub_stream4 = stream.GetOrCreateSubStream(); in TEST_F()105 Stream* sub_stream1 = stream.GetOrCreateSubStream(); in TEST_F()115 Stream* sub_stream2 = stream.GetOrCreateSubStream(); in TEST_F()132 Stream* sub_stream3 = stream.GetOrCreateSubStream(); in TEST_F()144 Stream* sub_stream1 = stream.GetOrCreateSubStream(); in TEST_F()[all …]
152 Stream *GetOrCreateSubStream() TF_LOCKS_EXCLUDED(mu_);
1290 Stream *Stream::GetOrCreateSubStream() { in GetOrCreateSubStream() function in stream_executor::Stream
55 se::Stream* substream = stream->GetOrCreateSubStream(); in TransferLiteralFromDevice()81 se::Stream* substream = stream->GetOrCreateSubStream(); in TransferLiteralFromDevice()105 se::Stream* substream = stream->GetOrCreateSubStream(); in TransferLiteralToDevice()121 se::Stream* substream = stream->GetOrCreateSubStream(); in TransferArrayFromDevice()149 se::Stream* substream = stream->GetOrCreateSubStream(); in TransferArrayToDevice()
372 se::Stream* deregister_stream = stream->GetOrCreateSubStream(); in UnregisterCancellation()
331 auto stream2 = stream_->GetOrCreateSubStream(); in XLA_TEST_F()
176 ? device_to_device_master_stream->GetOrCreateSubStream() in TpuDeviceToDeviceCopy()