516f97c543f0816677fd00ec24b47f31f2618f67,doc/source/notebooks/tailor/gp_nn.pct.py,,map_fn,#Any#,47

Before Change

def map_fn(input_slice: Dict[str, tf.Tensor]):
    updated = input_slice
    image = tf.cast(updated["image"], default_float()) / 255.
    label = tf.cast(updated["label"], default_float())
    return tf.reshape(image, [-1, image_size]), label

autotune = tf.data.experimental.AUTOTUNE
dataset = original_dataset\

After Change

def map_fn(input_slice: Dict[str, tf.Tensor]):
    updated = input_slice
    image = to_default_float(updated["image"]) / 255.
    label = to_default_float(updated["label"])
    return tf.reshape(image, [-1, image_size]), label

autotune = tf.data.experimental.AUTOTUNE
dataset = original_dataset\

Instances