ashutosh-arm commented on a change in pull request #9682:
URL: https://github.com/apache/tvm/pull/9682#discussion_r766744080
##########
File path: tests/python/contrib/test_cmsisnn/utils.py
##########
@@ -86,18 +86,20 @@ def make_module(func):
return mod
-def get_same_padding(data, kernel, dilation, stride, cmsisnn_padding=True):
+def get_padding(data, kernel, dilation, stride, padding):
"""Provides CMSIS-NN padding when output dim == input dim"""
Review comment:
CMSIS-NN padding is same as TFLU padding. This function was for
providing "SAME" kind of padding defined by TFL / TF specs. When we expect
output to be of the same dim as input, input to Conv2D / other ops needs to be
padded first.
--
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]
