-
Asim Shankar authored
This is a step towards merging the example in https://github.com/tensorflow/tpu-demos/tree/master/cloud_tpu/models/mnist with this repository, so we have a single model definition for training across CPU/GPU/eager execution/TPU. The change to dataset.py is so that the raw data can be read from cloud storage systems (like GCS and S3).
7087ec9b