Import rename for Estimator modules.

PiperOrigin-RevId: 219793067

Author: Michael Case

examples/toy_model_tpu.py

@@ -26,9 +26,9 @@
 from tensorflow.contrib.tpu.python.tpu import tpu_config
 from tensorflow.contrib.tpu.python.tpu import tpu_estimator
 from tensorflow.python.data.ops.dataset_ops import Dataset
-from tensorflow.python.estimator import estimator as estimator_lib
 from tensorflow.python.platform import flags
 from tensorflow.python.platform import tf_logging as logging
+from tensorflow_estimator.python.estimator import estimator as estimator_lib
 
 
 FLAGS = flags.FLAGS