Slow execution of TensorFlow 2.0. Update only a part of tf.Variable

Question

I am trying to translate a code from TensorFlow 1x to TensorFlow 2x. The code is an implementation of Word2Vec. The code is running properly but the execution is twice as sl