set_attrs_type_key added

apache · Oct 21, 2018 · 15b6a24 · 15b6a24
1 parent cea5add
commit 15b6a24
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/relay/op/tensor/transform.cc b/src/relay/op/tensor/transform.cc
@@ -830,6 +830,7 @@ Examples::
 .set_num_inputs(1)
 .add_argument("data", "Tensor", "The input tensor.")
 .set_support_level(4)
+.set_attrs_type_key("relay.attrs.StridedSliceAttrs")
 .add_type_rel("StridedSlice", StridedSliceRel);
 
 }  // namespace relay

diff --git a/tests/python/relay/test_op_level4.py b/tests/python/relay/test_op_level4.py
@@ -84,7 +84,6 @@ def verify_strided_slice(data, begin, end, stride, output):
     assert "end=" in z.astext()
     if stride:
         assert "stride=" in z.astext()
-    print(zz.checked_type)
     assert zz.checked_type == relay.ty.TensorType(output, "float32")
 
 def test_strided_slice():
@@ -105,4 +104,4 @@ def test_strided_slice():
     test_where()
     test_arg_reduce()
     test_multibox_prior()
-   test_strided_slice()
+    test_strided_slice()