fix missing ffi binding of relay.attrs.DequantizeAttrs (#7054)

apache · Dec 8, 2020 · 8ac40fa · 8ac40fa
1 parent 3144cec
commit 8ac40fa
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/python/tvm/relay/op/op_attrs.py b/python/tvm/relay/op/op_attrs.py
@@ -469,6 +469,11 @@ class QuantizeAttrs(Attrs):
     """Attributes used in quantize operators"""
 
 
+@tvm._ffi.register_object("relay.attrs.DequantizeAttrs")
+class DequantizeAttrs(Attrs):
+    """Attributes used in dequantize operators"""
+
+
 @tvm._ffi.register_object("relay.attrs.ReduceAttrs")
 class ReduceAttrs(Attrs):
     """Attributes used in reduction operators (e.g. sum)"""