Only apply check for non-tensor case

PiperOrigin-RevId: 261408908
2019-08-02 16:00:01 -07:00 · 2019-08-02 16:00:01 -07:00 · 9c14f6ba30
commit 9c14f6ba30
parent 68ea31c83c
1 changed files with 12 additions and 10 deletions
--- a/tensorflow/python/keras/backend.py
+++ b/tensorflow/python/keras/backend.py
@ -4031,6 +4031,8 @@ def in_train_phase(x, alt, training=None):
  if training is None:
    training = learning_phase()

+  # TODO(b/138862903): Handle the case when training is tensor.
+  if not tensor_util.is_tensor(training):
    if training == 1 or training is True:
      if callable(x):
        return x()