[TPUEstimator] Automatically detect the TPU system information, including topology...
authorJianwei Xie <xiejw@google.com>
Sun, 11 Feb 2018 23:54:39 +0000 (15:54 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Mon, 12 Feb 2018 00:01:11 +0000 (16:01 -0800)
commit30bb88dba14d7bfb0472bd79f949014b6f2902a7
tree8cc230714d825edbc2984a63da1953a70a531ec6
parent695ae5a3de96069c58a4041faa05fc4f68825035
[TPUEstimator] Automatically detect the TPU system information, including topology for model parallelism.

PiperOrigin-RevId: 185318852
tensorflow/contrib/tpu/BUILD
tensorflow/contrib/tpu/python/tpu/tpu_config.py
tensorflow/contrib/tpu/python/tpu/tpu_config_test.py
tensorflow/contrib/tpu/python/tpu/tpu_context.py [new file with mode: 0644]
tensorflow/contrib/tpu/python/tpu/tpu_estimator.py
tensorflow/contrib/tpu/python/tpu/tpu_system_metadata.py [new file with mode: 0644]