ekalda commented on a change in pull request #9209:
URL: https://github.com/apache/tvm/pull/9209#discussion_r725076965
##########
File path: python/tvm/relay/op/contrib/ethosu.py
##########
@@ -253,10 +255,52 @@ def is_valid(self) -> bool:
legal_groups = [1, self.ofm.shape[3]]
if self.groups not in legal_groups:
return False
- # This should be a valid QnnDepthwise2DParams, not QnnConv2DParams
+ # This should be a valid QnnDepthwiseConv2DParams, not QnnConv2DParams
return not self.is_depthwise
+class QnnDepthwiseConv2DParams(QnnConv2DParams):
+ """
+ This class will parse a call to a ethosu.depthwise_conv2d composite
function
+ and extract the parameter information.
+ """
+
+ composite_name = "ethosu.depthwise_conv2d"
+ # The hardware only supports padding upto the numbers as follows
+ padding_bounds = [31, 31, 32, 32]
+
+ def __init__(self, func_body):
Review comment:
Done
--
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]
