Skip to content

Commit 8a6e540

Browse files
committed
Do not drop DQ and Q (16 bit) in DQ -> DepthToSpace -> Q
1 parent c36dea7 commit 8a6e540

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

onnxruntime/core/optimizer/qdq_transformer/selectors_actions/qdq_selector_action_transformer.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,8 @@ void DropQDQNodesRules(SelectorActionRegistry& qdq_selector_action_registry) {
7777
true,
7878
cpu_ep);
7979
qdq_selector_action_registry.RegisterSelectorAndAction(drop_action_no_int16_name,
80-
{{"Resize", {}}},
80+
{{"DepthToSpace", {}},
81+
{"Resize", {}}},
8182
std::move(selector_no_16bit),
8283
std::move(drop_action_no_int16));
8384

@@ -94,8 +95,7 @@ void DropQDQNodesRules(SelectorActionRegistry& qdq_selector_action_registry) {
9495
// SpaceToDepth not included because there are no integer implementations.
9596
// https://github.com/microsoft/onnxruntime/issues/21287
9697
qdq_selector_action_registry.RegisterSelectorAndAction(drop_action_name,
97-
{{"DepthToSpace", {}},
98-
{"Expand", {}},
98+
{{"Expand", {}},
9999
{"Flatten", {}},
100100
{"Gather", {}},
101101
{"GatherElements", {}},

0 commit comments

Comments
 (0)