From bae9462c86861fcfe9402b37eade92eb8a69877d Mon Sep 17 00:00:00 2001 From: Yundi Qian Date: Fri, 4 Feb 2022 08:16:40 +0000 Subject: [PATCH] tf.nn.relu6 is already register, remove it from gin registration --- compiler_opt/rl/gin_external_configurables.py | 1 - 1 file changed, 1 deletion(-) diff --git a/compiler_opt/rl/gin_external_configurables.py b/compiler_opt/rl/gin_external_configurables.py index 34ae60d2..c2593092 100644 --- 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')