Searched refs:old_cuda_device (Results 1 – 1 of 1) sorted by relevance
811 absl::optional<int> old_cuda_device = absl::nullopt; in ConvertAfterShapes() local820 old_cuda_device = cuda_device_id; in ConvertAfterShapes()824 auto restore_cuda_device = gtl::MakeCleanup([old_cuda_device] { in ConvertAfterShapes()825 if (old_cuda_device.has_value()) { in ConvertAfterShapes()826 cudaSetDevice(old_cuda_device.value()); in ConvertAfterShapes()