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

Function identity_block_bn

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

Identity block with no shortcut convolution

(input_tensor)

Source from the content-addressed store, hash-verified

177
178
179def identity_block_bn(input_tensor):
180 """
181 Identity block with no shortcut convolution
182 """
183 y = tf.keras.layers.Conv2D(filters=12, kernel_size=(3, 3), padding="same")(
184 input_tensor
185 )
186 y = relu_bn(y)
187 y = tf.keras.layers.Conv2D(filters=24, kernel_size=(3, 3), padding="same")(y)
188 y = bn(y)
189 out = tf.keras.layers.Add()([y, input_tensor])
190 out = relu(out)
191 return out
192
193
194def identity_block_short_conv_bn(input_tensor):

Callers 1

pippin_28_28Function · 0.85

Calls 3

relu_bnFunction · 0.85
bnFunction · 0.85
reluFunction · 0.70

Tested by

no test coverage detected