4https://github.com/tensorflow/tpu/tree/master/models/official/resnet