tf.compat.v1.assign_add