Home
last modified time | relevance | path

Searched refs:ReturnSubStream (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_test.cc55 stream.ReturnSubStream(sub_stream1); in TEST_F()
60 stream.ReturnSubStream(sub_stream1); in TEST_F()
84 stream.ReturnSubStream(sub_stream1); in TEST_F()
91 stream.ReturnSubStream(sub_stream2); in TEST_F()
114 stream.ReturnSubStream(sub_stream1); in TEST_F()
131 stream.ReturnSubStream(sub_stream2); in TEST_F()
146 stream.ReturnSubStream(sub_stream1); in TEST_F()
196 stream.ReturnSubStream(sub_stream2); in TEST_F()
Dstream.h144 void ReturnSubStream(Stream *sub_stream) LOCKS_EXCLUDED(mu_);
Dstream.cc1785 void Stream::ReturnSubStream(Stream *sub_stream) { in ReturnSubStream() function in stream_executor::Stream
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.cc55 [&]() { stream->ReturnSubStream(substream); }); in TransferLiteralFromDevice()
80 [&]() { stream->ReturnSubStream(substream); }); in TransferLiteralFromDevice()
105 [&]() { stream->ReturnSubStream(substream); }); in TransferLiteralToDevice()
120 [&]() { stream->ReturnSubStream(substream); }); in TransferArrayFromDevice()
148 [&]() { stream->ReturnSubStream(substream); }); in TransferArrayToDevice()