MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / sam_32_32

Function sam_32_32

tools/tensorflow-quantization/tests/network_pool.py:113–130  ·  view source on GitHub ↗

Dummy network with resnet like architecture.

()

Source from the content-addressed store, hash-verified

111
112
113def sam_32_32():
114 """
115 Dummy network with resnet like architecture.
116 """
117 input_img = tf.keras.layers.Input(shape=(32, 32, 3))
118 x = tf.keras.layers.Conv2D(filters=12, kernel_size=(3, 3))(input_img)
119 x = tf.keras.layers.ReLU()(x)
120 x = tf.keras.layers.Conv2D(filters=24, kernel_size=(3, 3))(x)
121 x = tf.keras.layers.ReLU()(x)
122 x = identity_block_plain(x)
123 x = identity_block_plain(x)
124 x = identity_block_short_conv_plain(x)
125 x = tf.keras.layers.MaxPooling2D(pool_size=(2, 2))(x)
126 x = tf.keras.layers.Flatten()(x)
127 x = tf.keras.layers.Dense(100)(x)
128 x = tf.keras.layers.ReLU()(x)
129 x = tf.keras.layers.Dense(10)(x)
130 return tf.keras.Model(input_img, x, name="Sam")
131
132
133##############################################

Callers 2

test_keras_travellerFunction · 0.90
test_convert_to_onnxFunction · 0.90

Calls 2

identity_block_plainFunction · 0.85

Tested by 2

test_keras_travellerFunction · 0.72
test_convert_to_onnxFunction · 0.72