Home
last modified time | relevance | path

Searched refs:ReturnSubStream (Results 1 – 6 of 6) 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.h158 void ReturnSubStream(Stream *sub_stream) TF_LOCKS_EXCLUDED(mu_);
Dstream.cc1342 void Stream::ReturnSubStream(Stream *sub_stream) { in ReturnSubStream() function in stream_executor::Stream
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.cc58 [&]() { stream->ReturnSubStream(substream); }); in TransferLiteralFromDevice()
83 [&]() { stream->ReturnSubStream(substream); }); in TransferLiteralFromDevice()
108 [&]() { stream->ReturnSubStream(substream); }); in TransferLiteralToDevice()
123 [&]() { stream->ReturnSubStream(substream); }); in TransferArrayFromDevice()
151 [&]() { stream->ReturnSubStream(substream); }); in TransferArrayToDevice()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_tpu_device.cc182 device_to_device_master_stream->ReturnSubStream( in TpuDeviceToDeviceCopy()
286 device_to_device_master_stream->ReturnSubStream( in TpuDeviceToDeviceCopy()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc407 stream->ReturnSubStream(deregister_stream); in UnregisterCancellation()