Skip to content

Commit 332f205

Browse files
mcr229pytorchmergebot
authored andcommitted
[XNNPACK][QS8] torch.nn.ELU (#104307)
Differential Revision: [D47075933](https://our.internmc.facebook.com/intern/diff/D47075933/) Pull Request resolved: #104307 Approved by: https://github.com/digantdesai
1 parent c4e084e commit 332f205

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

torch/ao/quantization/backend_config/executorch.py

+2
Original file line numberDiff line numberDiff line change
@@ -346,6 +346,7 @@ def _get_share_qparams_ops_configs() -> List[BackendPatternConfig]:
346346
]
347347
share_qparams_ops = [
348348
F.adaptive_avg_pool2d,
349+
F.elu,
349350
F.hardtanh,
350351
F.max_pool2d,
351352
F.pad,
@@ -355,6 +356,7 @@ def _get_share_qparams_ops_configs() -> List[BackendPatternConfig]:
355356
F.leaky_relu_,
356357
torch.nn.AdaptiveAvgPool2d,
357358
torch.nn.ConstantPad2d,
359+
torch.nn.ELU,
358360
torch.nn.MaxPool2d,
359361
torch.nn.ReLU6,
360362
torch.nn.Hardtanh,

0 commit comments

Comments
 (0)