tensorflow_estimator/python/estimator/model_fn.py
tensorflow_estimator/python/estimator/tpu/tpu_estimator.py

tensorflow_estimator/python/estimator/tpu/util.py
tensorflow_estimator/python/estimator/util.py

tensorflow_estimator/python/estimator/estimator.py
tensorflow_estimator/python/estimator/tpu/tpu_estimator.py