def METHOD_NAME():
    model = tf.keras.Sequential(
        [
            tf.keras.layers.Input(shape=(5, 5, 1)),
            tf.keras.layers.Conv2D(2, 3, name="layer1"),
            tf.keras.layers.BatchNormalization(),
            tf.keras.layers.Dense(4, name="layer2"),
        ]
    )
    return model