Drop QDQ nodes around Expland, Flatten and Tile.

CPU EP supports 8-bit data and the ops are data movement only. Makes handling of ops in https://github.com/microsoft/onnxruntime/blob/2580d935cbecd756cef435fb173a2f10237e9d2a/onnxruntime/core/optimizer/qdq_transformer/selectors_actions/shared/utils.cc#L34-L44 consistent. #21375
microsoft · Jul 24, 2024 · b081998 · b081998
1 parent 2580d93
commit b081998
Showing 1 changed file with 4 additions and 1 deletion.
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
@@ -61,8 +61,11 @@ void DropQDQNodesRules(SelectorActionRegistry& qdq_selector_action_registry) {
 
   std::unique_ptr<NodeSelector> selector = std::make_unique<QDQ::DropQDQNodesSelector>(true);
   qdq_selector_action_registry.RegisterSelectorAndAction(drop_action_name,
-                                                         {{"Gather", {}},
+                                                         {{"Expand", {}},
+                                                          {"Flatten", {}},
+                                                          {"Gather", {}},
                                                           {"Reshape", {}},
+                                                          {"Tile", {}},
                                                           {"Transpose", {}},
                                                           {"Squeeze", {}},
                                                           {"Unsqueeze", {}}},