Searched refs:BorrowStreamFromPool (Results 1 – 3 of 3) sorted by relevance
122 std::unique_ptr<se::Stream> BorrowStreamFromPool();
122 std::unique_ptr<se::Stream> LocalDeviceState::BorrowStreamFromPool() { in BorrowStreamFromPool() function in xla::LocalDeviceState
1073 stream = local_device_state->BorrowStreamFromPool(); in Release()1098 block_stream = local_device_state->BorrowStreamFromPool(); in Release()1455 stream = local_device_state->BorrowStreamFromPool(); in BlockHostUntilReady()