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

Function merry_28_28

tools/tensorflow-quantization/tests/network_pool.py:247–260  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

245
246
247def merry_28_28():
248 input_img = tf.keras.layers.Input(shape=(28, 28))
249 x = tf.keras.layers.Reshape(target_shape=(28, 28, 1))(input_img)
250 x = tf.keras.layers.Conv2D(filters=8, kernel_size=(3, 3))(x)
251 x = relu_bn(x)
252 x = tf.keras.layers.Conv2D(filters=12, kernel_size=(3, 3))(x)
253 x = relu_bn(x)
254 x = inception_block(x)
255 x = tf.keras.layers.MaxPooling2D(pool_size=(2, 2))(x)
256 x = tf.keras.layers.Flatten()(x)
257 x = tf.keras.layers.Dense(100)(x)
258 x = tf.keras.layers.ReLU()(x)
259 x = tf.keras.layers.Dense(10)(x)
260 return tf.keras.Model(input_img, x, name="Merry")
261
262
263def pippin_28_28():

Calls 2

relu_bnFunction · 0.85
inception_blockFunction · 0.85