Searched refs:GetDeviceToHostStream (Results 1 – 3 of 3) sorted by relevance
114 se::Stream* GetDeviceToHostStream();
131 se::Stream* LocalDeviceState::GetDeviceToHostStream() { in GetDeviceToHostStream() function in xla::LocalDeviceState
596 auto* stream = local_device->GetDeviceToHostStream(); in logical_on_device_shape()1320 se::Stream* stream = local_device->GetDeviceToHostStream(); in ToLiteral()