tf.io.RaggedFeature.RowStarts

RowStarts(key,)