Tf.variable.assign