This is an automated email from the ASF dual-hosted git repository. zhic pushed a commit to branch master in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/master by this push: new bd78fb2 [RELAY][PY] Fix relay node registration after refactor (#5083) bd78fb2 is described below commit bd78fb28b82f32e6a83d5d983a580a84f4d0745c Author: Tianqi Chen <tqc...@users.noreply.github.com> AuthorDate: Tue Mar 17 10:59:23 2020 -0700 [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"""