Searched refs:ReturnSubStream (Results 1 – 6 of 6) sorted by relevance
55 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()
158 void ReturnSubStream(Stream *sub_stream) TF_LOCKS_EXCLUDED(mu_);
1342 void Stream::ReturnSubStream(Stream *sub_stream) { in ReturnSubStream() function in stream_executor::Stream
58 [&]() { 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()
182 device_to_device_master_stream->ReturnSubStream( in TpuDeviceToDeviceCopy()286 device_to_device_master_stream->ReturnSubStream( in TpuDeviceToDeviceCopy()
407 stream->ReturnSubStream(deregister_stream); in UnregisterCancellation()