in 5-appinfra/modules/hpc-ai-training-infra/tensorflow_mnist_train_distributed.py [0:0]
def scale(image, label): image = tf.cast(image, tf.float32) image /= 255 return image, label