Skip to content

Commit 85edf76

Browse files
committed
Minor fix based on PR review
1 parent 9869c8b commit 85edf76

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Deeploy/CommonExtensions/OptimizationPasses/TopologyOptimizationPasses/LoweringOptimizationPasses.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -307,6 +307,7 @@ def _NCWHtoNHWC_dw_fun(graph: gs.Graph, match: Match, name: str, default_channel
307307
node = next(iter((match.nodes_map.values())))
308308

309309
if not _isDepthwise(node):
310+
if opNode.attrs.get('group', 1) == 1:
310311
return graph
311312

312313
channels_first = node.attrs.get("channels_first", True)

0 commit comments

Comments
 (0)