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

Function lotho_28_28

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

Source from the content-addressed store, hash-verified

224
225
226def lotho_28_28():
227 input_img = tf.keras.layers.Input(shape=(28, 28), name="input_0")
228 r = tf.keras.layers.Reshape(target_shape=(28, 28, 1), name="reshape_0")(input_img)
229 x = tf.keras.layers.DepthwiseConv2D(kernel_size=(3, 3), name="dconv_0")(r)
230 x = tf.keras.layers.ReLU(name="relu_0")(x)
231 x = tf.keras.layers.DepthwiseConv2D(kernel_size=(3, 3), name="dconv_1")(x)
232 x = tf.keras.layers.ReLU(name="relu_1")(x)
233 x = tf.keras.layers.Flatten(name="flatten_0")(x)
234 return tf.keras.Model(input_img, x, name="Lotho")
235
236
237def lobelia_28_28():

Calls

no outgoing calls