add a comment about bailing early

apache · Feb 26, 2021 · 89b7dee · 89b7dee
1 parent 6f62717
commit 89b7dee
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/relay/transforms/simplify_expr.cc b/src/relay/transforms/simplify_expr.cc
@@ -132,6 +132,8 @@ class SimplifyConvPad : public SimplifyPattern {
     std::string data_layout = attrs->data_layout;
     std::set<char> image_dims({'H', 'W', 'D'});
     Array<PrimExpr> padding;
+    // If we're padding a non-spatial dimension, don't simplify
+    // Convolution can only pad on spatial axes
     for (size_t i = 0; i < param->pad_width.size(); ++i) {
       if (!image_dims.count(data_layout[i])) {
         for (size_t j = 0; j < param->pad_width[i].size(); ++j) {