tf.ragged.stack_dynamic_partitions