Searched refs:ReturnStream (Results 1 – 2 of 2) sorted by relevance
33 void operator()(se::Stream* stream) { pool->ReturnStream(stream); } in operator()56 void ReturnStream(se::Stream* stream);
55 void StreamPool::ReturnStream(se::Stream* stream) { in ReturnStream() function in xla::StreamPool