[RELAY][PY] Fix relay node registration after refactor (apache#5083)

neo-ai · Apr 17, 2020 · 4db67b8 · 4db67b8
1 parent fcf7de6
commit 4db67b8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/tvm/relay/op/op_attrs.py b/python/tvm/relay/op/op_attrs.py
@@ -43,10 +43,12 @@ class Conv2DWinogradWeightTransformAttrs(Attrs):
 class Conv2DWinogradNNPACKWeightTransformAttrs(Attrs):
     """Attributes for nn.contrib_conv2d_winograd_nnpack_weight_transform"""
 
-@register_relay_attr_node
+
+@tvm._ffi.register_object("relay.attrs.Dilation2DAttrs")
 class Dilation2DAttrs(Attrs):
     """Attributes for nn.dilation2d"""
 
+
 @tvm._ffi.register_object("relay.attrs.GlobalPool2DAttrs")
 class GlobalPool2DAttrs(Attrs):
     """Attributes for nn.global_pool"""