Fix for the broken --config=rocm build.

Note: This ia a different PR from PRs tensorflow#28189 and tensorflow#28263 (same symptom, different cause and fix)

The `--config=rocm` build was broken by the following commit.

tensorflow@481748b

The changes made by the above commit had one bug in the changes for the ROCm platform, whias was leading to the build failure. Fixing that bug to make the `--config=rocm` build working again.

tensorflow/stream_executor/rocm/rocm_driver.cc

@@ -305,8 +305,8 @@ static port::Status InternalInit() {
 /* static */ port::Status GpuDriver::Init() {
   // Cached return value from calling InternalInit(), as hipInit need only be
   // called once, but GpuDriver::Init may be called many times.
-  static port::Status* init_retval = [&] {
-    init_retval = new Status(InternalInit());
+  static port::Status* init_retval = [] {
+    return new port::Status(InternalInit());
   }();
   return *init_retval;
 }