import tensorflow as tf

      outputs = {
          'foo': tf.RaggedTensor.from_row_splits(
              values=tf.constant([3, 1, 4, 1, 5, 9, 2, 6], tf.int64),
              row_splits=[0, 4, 4, 7, 8, 8]),
