Skip to content

Commit bf5a286

Browse files
committed
Minor fix based on PR review
1 parent d1a3ce8 commit bf5a286

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Deeploy/CommonExtensions/OptimizationPasses/TopologyOptimizationPasses/LoweringOptimizationPasses.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -341,7 +341,7 @@ def _PULPDWNCHWtoNHWC_fun(graph: gs.Graph, match: Match, name: str, default_chan
341341
opNode = matched_nodes[0]
342342
node_op = opNode.op
343343

344-
if 'group' in opNode.attrs and opNode.attrs['group'] == 1:
344+
if opNode.attrs.get('group', 1) == 1:
345345
return graph
346346

347347
if (("channels_first" in opNode.attrs and opNode.attrs["channels_first"] != default_channels_first)

0 commit comments

Comments
 (0)