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

Function identity_block_plain

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

Identity block with no shortcut convolution

(input_tensor)

Source from the content-addressed store, hash-verified

58###### Tiny, ResNet like network ####################
59#####################################################
60def identity_block_plain(input_tensor):
61 """
62 Identity block with no shortcut convolution
63 """
64 y = tf.keras.layers.Conv2D(filters=12, kernel_size=(3, 3), padding="same")(
65 input_tensor
66 )
67 y = tf.keras.layers.ReLU()(y)
68 y = tf.keras.layers.Conv2D(filters=24, kernel_size=(3, 3), padding="same")(y)
69 out = tf.keras.layers.Add()([y, input_tensor])
70 out = tf.keras.layers.ReLU()(out)
71 return out
72
73
74def identity_block_short_conv_plain(input_tensor):

Callers 2

frodo_32_32Function · 0.85
sam_32_32Function · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected