diff --git a/python/tvm/relay/op/transform.py b/python/tvm/relay/op/transform.py index 1826a51a9a89c..aee84805facb7 100644 --- 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 index 7153ef112a203..fef0302a05074 100644 --- 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>();