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 <[email protected]>
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"""
