提交 53766c69 编辑于 作者: Keith Rush's avatar Keith Rush 提交者: tensorflow-copybara
浏览文件

Re-enables tf.function in keras_utils' predict_on_batch conversion.

PiperOrigin-RevId: 394111901
上级 9885f13b
......@@ -334,6 +334,7 @@ class _KerasModel(model_lib.Model):
def input_spec(self):
return self._input_spec
@tf.function
def predict_on_batch(self, x, training=True):
return self._keras_model(x, training=training)
......
支持 Markdown
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册