Function frodo_32_32

tools/tensorflow-quantization/tests/network_pool.py:93–110

Dummy network with resnet like architecture.

()

def frodo_32_32():
    """
    Dummy network with resnet like architecture.
    """
    input_img = tf.keras.layers.Input(shape=(32, 32, 3))
    x = tf.keras.layers.Conv2D(filters=12, kernel_size=(3, 3))(input_img)
    x = tf.keras.layers.ReLU()(x)
    x = tf.keras.layers.Conv2D(filters=24, kernel_size=(3, 3))(x)
    x = tf.keras.layers.ReLU()(x)
    x = tf.keras.layers.MaxPooling2D(pool_size=(2, 2))(x)
    x = identity_block_plain(x)
    x = identity_block_short_conv_plain(x)
    x = tf.keras.layers.MaxPooling2D(pool_size=(2, 2))(x)
    x = tf.keras.layers.Flatten()(x)
    x = tf.keras.layers.Dense(100)(x)
    x = tf.keras.layers.ReLU()(x)
    x = tf.keras.layers.Dense(10)(x)
    return tf.keras.Model(input_img, x, name="Frodo")


def sam_32_32():