From 17127c21de1a61a0c0c15b710af85e62e8dbf0a7 Mon Sep 17 00:00:00 2001 From: Nick Hynes Date: Mon, 18 Mar 2019 21:33:02 +0000 Subject: [PATCH] Use existing new_var function --- python/tvm/relay/frontend/keras.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/relay/frontend/keras.py b/python/tvm/relay/frontend/keras.py index 48b27009b9998..8810b1d07746c 100644 --- a/python/tvm/relay/frontend/keras.py +++ b/python/tvm/relay/frontend/keras.py @@ -7,7 +7,7 @@ from .. import expr as _expr from .. import op as _op from ... import nd as _nd -from .common import ExprTable +from .common import ExprTable, new_var __all__ = ['from_keras'] @@ -656,7 +656,7 @@ def from_keras(model, shape=None): def _convert_input_layer(keras_layer): input_name = keras_layer.name input_shape = shape[input_name] if shape is not None and input_name in shape else None - etab.set_expr(input_name, _expr.var(input_name, shape=input_shape)) + etab.set_expr(input_name, new_var(input_name, shape=input_shape)) etab = ExprTable() for keras_layer in model.layers: