Searched refs:ReturnStreamToPool (Results 1 – 3 of 3) sorted by relevance
125 void ReturnStreamToPool(std::unique_ptr<se::Stream> stream);
140 void LocalDeviceState::ReturnStreamToPool(std::unique_ptr<se::Stream> stream) { in ReturnStreamToPool() function in xla::LocalDeviceState
1080 local_device_state->ReturnStreamToPool(std::move(stream)); in Release()1108 local_device_state->ReturnStreamToPool( in Release()1462 local_device_state->ReturnStreamToPool(std::move(stream)); in BlockHostUntilReady()