def scale()

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