Skip to content

Commit 7b5448f

Browse files
authored
Fix num nonbatch dims calculation
Differential Revision: D73967842 Pull Request resolved: #10600
1 parent e8b96c6 commit 7b5448f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

backends/xnnpack/operators/quant_params.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,8 @@ def _feeds_into_conv(node: torch.fx.Node) -> bool:
149149

150150
while users_list:
151151
current_user = users_list.pop()
152+
if "linear" in str(current_user.target):
153+
return False
152154
if "convolution" in str(current_user.target):
153155
return True
154156
users_list.extend(current_user.users)

0 commit comments

Comments
 (0)