Home
last modified time | relevance | path

Searched refs:GetOrCreateSubStream (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_test.cc53 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 …]
Dstream.h152 Stream *GetOrCreateSubStream() TF_LOCKS_EXCLUDED(mu_);
Dstream.cc1290 Stream *Stream::GetOrCreateSubStream() { in GetOrCreateSubStream() function in stream_executor::Stream
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.cc55 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()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc372 se::Stream* deregister_stream = stream->GetOrCreateSubStream(); in UnregisterCancellation()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtransfer_manager_test.cc331 auto stream2 = stream_->GetOrCreateSubStream(); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_tpu_device.cc176 ? device_to_device_master_stream->GetOrCreateSubStream() in TpuDeviceToDeviceCopy()