def func(x):
      rt1 = RaggedTensor.from_row_splits(
          values=x, row_splits=[0, 0, 0, 0])
      return map_fn.map_fn(
          math_ops.reduce_max,