diff --git a/src/relay/transforms/fold_explicit_padding.cc b/src/relay/transforms/fold_explicit_padding.cc index 794bcfd3d07d..55e1fe854fe3 100644 --- a/src/relay/transforms/fold_explicit_padding.cc +++ b/src/relay/transforms/fold_explicit_padding.cc @@ -228,7 +228,6 @@ class SimplifyExplicitPad { const Expr& pv = pad_node->args[1]; const ConstantNode* pad_value = pv.as(); - auto pad_scalar = ToScalar(pad_value->data); if (node_map.find(qconv2d_) != node_map.end()) { Attrs attrs = MakeConv2D3DAttrs(param, call_node->attrs.as()); @@ -251,6 +250,7 @@ class SimplifyExplicitPad { if (param->pad_mode == "constant" && pad_value) { Attrs attrs; + auto pad_scalar = ToScalar(pad_value->data); if (pad_scalar == 0.0) { // Fold Padding and Conv/AvgPool only if pad_value == 0. if (node_map.count(conv_)) {