ashutosh-arm commented on a change in pull request #9409:
URL: https://github.com/apache/tvm/pull/9409#discussion_r752178429
##########
File path: src/relay/backend/contrib/cmsisnn/tir_to_runtime.cc
##########
@@ -88,8 +88,12 @@ class CodeGenCMSISNN : public codegen::CodeGenCHost {
std::string EmitCMSISNNConvParams(std::ostream& os, int32_t input_offset,
int32_t output_offset,
int32_t stride_w, int32_t stride_h,
int32_t padding_w,
int32_t padding_h, int32_t dilation_w,
int32_t dilation_h,
- int32_t clip_min, int32_t clip_max) {
- std::string struct_name = "conv_params";
+ int32_t clip_min, int32_t clip_max,
int32_t depth_multiplier) {
+ std::string struct_name = "cmsis_nn_conv_params";
+ std::string instance_name = "conv_params";
+ if (depth_multiplier != -1) {
+ struct_name = "cmsis_nn_dw_conv_params";
+ }
Review comment:
ACK
--
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]
