jwfromm commented on code in PR #15587:
URL: https://github.com/apache/tvm/pull/15587#discussion_r1298826085
##########
python/tvm/relax/frontend/nn/op.py:
##########
@@ -179,6 +179,89 @@ def matmul(a: Tensor, b: Tensor, out_dtype: Optional[str]
= None, name: str = "m
return _wrap_nested(_op.matmul(a._expr, b._expr, out_dtype=out_dtype),
name)
+def conv1d(
+ input: Tensor,
Review Comment:
Although i originally used this name for conv2d as well, I decided to switch
to `x` for consistency since thats what every other op does. We should probably
do the same here.
--
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]
