code: def scale(image, label):
      image = tf.cast(image, tf.float32)
      image /= 255
  
      return image, label
  
file_path: cloud/src/python/tensorflow_cloud/core/tests/testdata/mnist_example_using_fit.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 1 for calling a global function with example usage: scale(image, label) and returns: image,  label