code: def get_cnn_model(features):
      """Given input features, returns the logits from a simple CNN model."""
      input_layer = tf.reshape(features, [-1, 28, 28, 1])
      y = tf.keras.layers.Conv2D(
          16, 8, strides=2, padding="same", activation="relu"
      )(input_layer)
      y = tf.keras.layers.MaxPool2D(2, 1)(y)
      y = tf.keras.layers.Conv2D(
          32, 4, strides=2, padding="valid", activation="relu"
      )(y)
      y = tf.keras.layers.MaxPool2D(2, 1)(y)
      y = tf.keras.layers.Flatten()(y)
      y = tf.keras.layers.Dense(32, activation="relu")(y)
      logits = tf.keras.layers.Dense(10)(y)
  
      return logits
  
label_desc: Use image node_id 1 for calling a global function with example usage: get_cnn_model(features) and returns: logits