Common tags used for graphs in SavedModel.
GPU = 'gpu'
SERVING = 'serve'
TPU = 'tpu'
TRAINING = 'train'