Reducing Batch-size for shapemask while running on Cloud TPU v3-8. (#559)
Reducing Batch-size for shapemask while running on Cloud TPU v3-8. (#559)

* Reducing Batch-size for shapemask while running on Cloud TPU v3-8.
GoogleCloudPlatform/ml-testing-accelerators,GoogleCloudPlatform/ml-testing-accelerators