masahi commented on a change in pull request #4639: [Relay/Topi][Op] Conv1D
URL: https://github.com/apache/incubator-tvm/pull/4639#discussion_r364752270
##########
File path: topi/python/topi/generic/nn.py
##########
@@ -34,6 +34,42 @@ def _default_schedule(outs, auto_inline):
return s
+@tvm.target.generic_func
+def schedule_conv1d_ncw(outs):
+ """Schedule for conv1d_ncw
+
+ Parameters
+ ----------
+ outs: Array of Tensor
+ The computation graph description of conv2d_hwcn
+ in the format of an array of tensors.
+
+ Returns
+ -------
+ sch: Schedule
+ The computation schedule for the op.
+ """
+ return _default_schedule(outs, False)
+
+
+@tvm.target.generic_func
+def schedule_conv1d_nwc(outs):
+ """Schedule for conv1d_nwc
+
+ Parameters
+ ----------
+ outs: Array of Tensor
+ The computation graph description of conv2d_hwcn
Review comment:
needs doc fix
----------------------------------------------------------------
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.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
