code: def _build_model():
      """Builds model with MyDenseLayer."""
      i = tf.keras.layers.Input(shape=(28, 28), name="input")
      x = tf.keras.layers.Reshape((28, 28, 1))(i)
      x = tf.keras.layers.Conv2D(
          filters=12,
          kernel_size=(3, 3),
          activation="relu",
          name="conv1",
      )(x)
      x = tf.keras.layers.MaxPool2D(2, 2)(x)
      x = tf.keras.layers.Flatten()(x)
      output = MyDenseLayer(units=10)(x)
  
      model = tf.keras.Model(inputs=[i], outputs=[output])
      return model
  
imports: import tensorflow
  from tensorflow_model_optimization.python.core.clustering.keras import cluster
  from tensorflow_model_optimization.python.core.clustering.keras import cluster_config
  from tensorflow_model_optimization.python.core.clustering.keras import clusterable_layer
  from tensorflow_model_optimization.python.core.clustering.keras import clustering_algorithm
label_desc: Use image node_id 1 for calling a global function with example usage: _build_model() and returns: model