Use existing new_var function

apache · Mar 18, 2019 · 17127c2 · 17127c2
1 parent 1b6caf4
commit 17127c2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: