Updated resolver code because of internal TensorFlow 1.x changes.
Updated resolver code because of internal TensorFlow 1.x changes.

PiperOrigin-RevId: 317889788

tensorflow/tpu,tensorflow/tpu,tensorflow/tpu,tensorflow/tpu