Skip to content

Commit 43fb017

Browse files
committed
[QUANTIZE] Fix.
1 parent fb4955a commit 43fb017

File tree

2 files changed

+2
-5
lines changed

2 files changed

+2
-5
lines changed

python/tvm/relay/op/transform.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -388,5 +388,5 @@ def slice_like(data, shape_like, axes=None):
388388
"""
389389
return _make.slice_like(data, shape_like, axes)
390390

391-
def simulated_quantize(data, scale, bit, sign=True, rounding='round'):
392-
return _make.simulated_quantize(data, scale, bit, sign, rounding)
391+
def simulated_quantize(data, scale, bit, clip_min, clip_max, sign=True, rounding='round'):
392+
return _make.simulated_quantize(data, scale, bit, clip_min, clip_max, sign, rounding)

src/relay/op/tensor/unary.cc

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -93,9 +93,6 @@ RELAY_REGISTER_UNARY_OP("relay.op._make.", "copy")
9393
// relay.clip
9494
TVM_REGISTER_NODE_TYPE(ClipAttrs);
9595

96-
// relay.clip
97-
TVM_REGISTER_NODE_TYPE(ClipAttrs);
98-
9996
TVM_REGISTER_API("relay.op._make.clip")
10097
.set_body_typed<Expr(Expr, double, double)>([](Expr a, double a_min, double a_max) {
10198
auto attrs = make_node<ClipAttrs>();

0 commit comments

Comments
 (0)