Lyken17 commented on pull request #9465: URL: https://github.com/apache/tvm/pull/9465#issuecomment-966434464
@vinx13 I've addressed the issue you mentioned and lint the file. The code should be ready to merge and feel free to point out if there is any placed needed to be updated. For test cases, https://github.com/apache/tvm/blob/main/tests/python/relay/test_op_level2.py, I am afraid that tests may not be able to added yet. The shape check about relay is somehow buggy and might be related with https://github.com/apache/tvm/pull/9336 to fix. Currently ```python n, h, w, c = te.size_var("n"), 10, 10, 12 g = 4 x = relay.var("x", relay.TensorType((n, h, w, c), "float32")) w = relay.var("w", relay.TensorType((12, 16 // g, 5, 5), "float32")) y = relay.nn.conv2d_transpose(x, w, output_padding=(1, 1), channels=16, groups=g, data_layout="NHWC") yy = run_infer_type(y) assert yy.checked_type == relay.TensorType((n, 15, 15, 16), "float32"), yy.checked_type # FAILED tests/python/relay/test_op_level2.py::test_conv2d_transpose_infer_type - AssertionError: TensorType([{n|n>=0}, 15, 15, 4], float32) ``` -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
