tf.compat.v1.create_partitioned_variables