diff --git a/tensorflow/python/keras/BUILD b/tensorflow/python/keras/BUILD index 612a3855ff9..4e8fed1b4e3 100755 --- a/tensorflow/python/keras/BUILD +++ b/tensorflow/python/keras/BUILD @@ -236,6 +236,7 @@ py_library( ], srcs_version = "PY2AND3", deps = [ + ":activations", ":backend", ":losses", "//tensorflow/python:array_ops", diff --git a/tensorflow/python/keras/metrics.py b/tensorflow/python/keras/metrics.py index f05fb910a72..5b7019793e6 100644 --- a/tensorflow/python/keras/metrics.py +++ b/tensorflow/python/keras/metrics.py @@ -37,8 +37,8 @@ from tensorflow.python.framework import dtypes from tensorflow.python.framework import ops from tensorflow.python.framework import tensor_shape from tensorflow.python.framework import tensor_spec +from tensorflow.python.keras import activations from tensorflow.python.keras import backend as K -from tensorflow.python.keras.activations import sigmoid from tensorflow.python.keras.engine import base_layer from tensorflow.python.keras.engine import base_layer_utils from tensorflow.python.keras.engine import keras_tensor @@ -2134,7 +2134,7 @@ class AUC(Metric): label_weights = None if self.multi_label else self.label_weights if self._from_logits: - y_pred = sigmoid(y_pred) + y_pred = activations.sigmoid(y_pred) with ops.control_dependencies(deps): return metrics_utils.update_confusion_matrix_variables(