experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Only apply check for non-tensor case

Browse Source 
PiperOrigin-RevId: 261408908
This commit is contained in:
Yanhua Sun 2019-08-02 16:00:01 -07:00 committed by TensorFlower Gardener
parent 68ea31c83c
commit 9c14f6ba30
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/keras/backend.py
View File

@ -4031,6 +4031,8 @@ def in_train_phase(x, alt, training=None):
if training is None: if training is None:
training = learning_phase() 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 training == 1 or training is True:
if callable(x): if callable(x):
return x() return x()