From bd78fb28b82f32e6a83d5d983a580a84f4d0745c Mon Sep 17 00:00:00 2001 From: Tianqi Chen Date: Tue, 17 Mar 2020 10:59:23 -0700 Subject: [PATCH] [RELAY][PY] Fix relay node registration after refactor (#5083) --- python/tvm/relay/op/op_attrs.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/tvm/relay/op/op_attrs.py b/python/tvm/relay/op/op_attrs.py index 53f3c63..4b71899 100644 --- 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""" -- 2.7.4