[TF:PJRT] Respect ExecuteOption.non_donatable_input_indices in GPU PJRT client.

Note TF runtime side is already set up in xla_launch_util.

PiperOrigin-RevId: 549456738

tensorflow/compiler/xla/pjrt/pjrt_stream_executor_client.cc

@@ -2288,7 +2288,10 @@ StatusOr<ScopedShapedBuffer> PjRtStreamExecutorExecutable::EnqueueExecution(
           "device %s, but replica is assigned to device %s.",
           i, replica, handle->device()->DebugString(), device->DebugString());
     }
-    bool must_donate = donate_it != donated_params.end() && *donate_it == i;
+    bool donation_denied_at_runtime =
+        options.non_donatable_input_indices.contains(i);
+    bool must_donate = donate_it != donated_params.end() && *donate_it == i &&
+                       !donation_denied_at_runtime;
     if (must_donate) {
       ++donate_it;
     }