Disables expensive validation in RaggedTensor factories.

They produce RaggedTensors from existing by changing values (but not values shape). The row splits will be valid as long as input RaggedTensors are valid.

PiperOrigin-RevId: 472779047

Author: iindyk

tensorflow_transform/mappers.py

@@ -1826,7 +1826,8 @@ def _make_composite_tensor_wrapper_if_composite(
   elif isinstance(x, tf.RaggedTensor):
 
     def from_nested_row_splits(values):
-      return tf.RaggedTensor.from_nested_row_splits(values, x.nested_row_splits)
+      return tf.RaggedTensor.from_nested_row_splits(
+          values, x.nested_row_splits, validate=False)
 
     return from_nested_row_splits
   else:

tensorflow_transform/tf_utils.py

@@ -470,7 +470,7 @@ def _to_string(x: common_types.TensorType) -> common_types.TensorType:
         dense_shape=x.dense_shape)
   elif isinstance(x, tf.RaggedTensor):
     return tf.RaggedTensor.from_row_splits(
-        values=_to_string(x.values), row_splits=x.row_splits)
+        values=_to_string(x.values), row_splits=x.row_splits, validate=False)
   else:
     return tf.strings.as_string(x)