[QUANTIZE] Fix.

apache · Nov 27, 2018 · 43fb017 · 43fb017
1 parent fb4955a
commit 43fb017
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/python/tvm/relay/op/transform.py b/python/tvm/relay/op/transform.py
@@ -388,5 +388,5 @@ def slice_like(data, shape_like, axes=None):
     """
     return _make.slice_like(data, shape_like, axes)
 
-def simulated_quantize(data, scale, bit, sign=True, rounding='round'):
-    return _make.simulated_quantize(data, scale, bit, sign, rounding)
+def simulated_quantize(data, scale, bit, clip_min, clip_max, sign=True, rounding='round'):
+    return _make.simulated_quantize(data, scale, bit, clip_min, clip_max, sign, rounding)
diff --git a/src/relay/op/tensor/unary.cc b/src/relay/op/tensor/unary.cc
@@ -93,9 +93,6 @@ RELAY_REGISTER_UNARY_OP("relay.op._make.", "copy")
 // relay.clip
 TVM_REGISTER_NODE_TYPE(ClipAttrs);
 
-// relay.clip
-TVM_REGISTER_NODE_TYPE(ClipAttrs);
-
 TVM_REGISTER_API("relay.op._make.clip")
   .set_body_typed<Expr(Expr, double, double)>([](Expr a, double a_min, double a_max) {
       auto attrs = make_node<ClipAttrs>();