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

Function otho_28_28

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

Source from the content-addressed store, hash-verified

213
214
215def otho_28_28():
216 input_img = tf.keras.layers.Input(shape=(28, 28), name="input_0")
217 r = tf.keras.layers.Reshape(target_shape=(28, 28, 1), name="reshape_0")(input_img)
218 x = tf.keras.layers.Conv2D(filters=2, kernel_size=(3, 3), name="conv_0")(r)
219 x = tf.keras.layers.ReLU(name="relu_0")(x)
220 x = tf.keras.layers.Conv2D(filters=2, kernel_size=(3, 3), name="conv_1")(x)
221 x = tf.keras.layers.ReLU(name="relu_1")(x)
222 x = tf.keras.layers.Flatten(name="flatten_0")(x)
223 return tf.keras.Model(input_img, x, name="Otho")
224
225
226def lotho_28_28():

Calls

no outgoing calls