tf.nn.relu6 is already register, remove it from gin registration

google · Feb 4, 2022 · bae9462 · bae9462
1 parent 5e16051
commit bae9462
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/compiler_opt/rl/gin_external_configurables.py b/compiler_opt/rl/gin_external_configurables.py
@@ -26,7 +26,6 @@
 # Activation function.
 config.external_configurable(tf.keras.activations.relu,
                              'tf.keras.activations.relu')
-config.external_configurable(tf.nn.relu6, 'tf.nn.relu6')
 
 # Layers
 config.external_configurable(tf.keras.layers.Add, 'tf.keras.layers.Add')