Do not drop DQ and Q (16 bit) in DQ -> DepthToSpace -> Q

microsoft · Jan 10, 2025 · 8a6e540 · 8a6e540
1 parent c36dea7
commit 8a6e540
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...ntime/core/optimizer/qdq_transformer/selectors_actions/qdq_selector_action_transformer.cc b/...ntime/core/optimizer/qdq_transformer/selectors_actions/qdq_selector_action_transformer.cc
@@ -77,7 +77,8 @@ void DropQDQNodesRules(SelectorActionRegistry& qdq_selector_action_registry) {
                                                                                                 true,
                                                                                                 cpu_ep);
   qdq_selector_action_registry.RegisterSelectorAndAction(drop_action_no_int16_name,
-                                                         {{"Resize", {}}},
+                                                         {{"DepthToSpace", {}},
+                                                          {"Resize", {}}},
                                                          std::move(selector_no_16bit),
                                                          std::move(drop_action_no_int16));
 
@@ -94,8 +95,7 @@ void DropQDQNodesRules(SelectorActionRegistry& qdq_selector_action_registry) {
   // SpaceToDepth not included because there are no integer implementations.
   // https://github.com/microsoft/onnxruntime/issues/21287
   qdq_selector_action_registry.RegisterSelectorAndAction(drop_action_name,
-                                                         {{"DepthToSpace", {}},
-                                                          {"Expand", {}},
+                                                         {{"Expand", {}},
                                                           {"Flatten", {}},
                                                           {"Gather", {}},
                                                           {"GatherElements", {}},