tf.dynamic_partition