code: def build_model():
      inputs = tf.keras.layers.Input(shape=(28, 28), name="input")
      x = tf.keras.layers.Reshape((28, 28, 1))(inputs)
      x = tf.keras.layers.Conv2D(
          20,
          5,
          use_bias=True,
          activation="relu",
          padding="valid",
          name="conv1",
      )(x)
      x = tf.keras.layers.MaxPool2D(2, 2)(x)
      x = tf.keras.layers.Conv2D(
          50,
          5,
          use_bias=True,
          activation="relu",
          padding="valid",
          name="conv2",
      )(x)
      x = tf.keras.layers.MaxPool2D(2, 2)(x)
      x = tf.keras.layers.Flatten()(x)
      x = tf.keras.layers.Dense(
          500, use_bias=True, activation="relu", name="fc1"
      )(x)
      outputs = tf.keras.layers.Dense(10, use_bias=True, name="fc2")(x)
      return tf.keras.Model(inputs=[inputs], outputs=[outputs])
  
imports: import os
  import shutil
  from absl.testing import parameterized
  import tensorflow
  from tensorflow_model_optimization.python.core.common.keras.compression.algorithms import epr
label_desc: Use image node_id 1 for calling a global function with example usage: build_model() and returns: tf