ScatterNdFunctor to update operands of all valid indices and contin…

…ue on bad indices.

This is to support the new attribute "bad_indices_policy". Passing downs the behavior also works, but it makes `ScatterNdFunctor` unnecessarily complicated while the only gain is the performance with out-of-bound error.

PiperOrigin-RevId: 637646021

tf2jax/_src/ops.py

@@ -1727,7 +1727,7 @@ def _func(
 @register_operation("ScatterNd")
 def _scatter_nd(proto):
   """Parse a ScatterNd op."""
-  _check_attrs(proto, {"T", "Tindices"})
+  _check_attrs(proto, {"T", "Tindices", "bad_indices_policy"})
 
   def _func(
       indices: jnp.ndarray,