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

Merge pull request #35019 from foxik:on_batch_methods_relax_shapes

Browse Source 
PiperOrigin-RevId: 285685827
Change-Id: Ic54760c3e6afeb1939b5c2a96e3d3694b57efbef
This commit is contained in:
TensorFlower Gardener 2019-12-15 18:02:36 -08:00
commit 08c45ed0c5
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/keras/engine/training_v2_utils.py
View File

@ -119,7 +119,9 @@ def _make_on_batch_function(model, mode):
func = model
if not model.run_eagerly:
func = def_function.function(func)
# Pass `experimental_relax_shapes` to avoid retracing for dynamic batch
# size, variable length sequences, etc.
func = def_function.function(func, experimental_relax_shapes=True)
return func