@@ -449,8 +449,8 @@ static bool HandleDoubleQDQ(onnxruntime::Graph& dst_graph, const onnxruntime::Gr
449
449
static void AddStandaloneNodeUnit (onnxruntime::Graph& dst_graph, const onnxruntime::GraphViewer& src_graph,
450
450
const NodeUnit& node_unit,
451
451
std::set<std::string>& initializers_to_keep,
452
- bool IsWeightSharingWithoutOVEPQDQStripping ,
453
- const logging::Logger& /* logger */ ) {
452
+ const logging::Logger& /* logger */ ,
453
+ bool IsWeightSharingWithoutOVEPQDQStripping ) {
454
454
assert (node_unit.UnitType () == NodeUnit::Type::SingleNode);
455
455
456
456
// this is the scenario where WAI is enabled and ovep stripping is disabled
@@ -521,8 +521,8 @@ static void AddQDQNodeUnit(onnxruntime::Graph& dst_graph,
521
521
const onnxruntime::GraphViewer& src_graph,
522
522
const NodeUnit& node_unit,
523
523
std::set<std::string>& initializers_to_keep,
524
- bool IsWeightSharingWithoutOVEPQDQStripping ,
525
- const logging::Logger& /* logger */ ) {
524
+ const logging::Logger& /* logger */ ,
525
+ bool IsWeightSharingWithoutOVEPQDQStripping ) {
526
526
assert (node_unit.UnitType () == NodeUnit::Type::QDQGroup);
527
527
528
528
// Collect inputs coming into the node unit.
@@ -683,7 +683,8 @@ Status CreateModelWithStrippedQDQNodes(const GraphViewer& src_graph,
683
683
bool enable_ovep_weight_sharing,
684
684
bool enable_ovep_qdq_optimizer,
685
685
/* out*/ std::unique_ptr<onnxruntime::Model>& model,
686
- /* out*/ sw& shared_weights) {
686
+ /* out*/ sw& shared_weights,
687
+ bool enable_ovep_qdq_optimizer) {
687
688
// NOTE: This function is a re-implementation of GraphViewerToProto() in core/graph/graph_proto_serializer.cc
688
689
// with the following differences:
689
690
// - Uses onnxruntime::Graph APIs instead of onnx::GraphProto APIs.
@@ -777,9 +778,9 @@ Status CreateModelWithStrippedQDQNodes(const GraphViewer& src_graph,
777
778
bool IsWeightSharingWithoutOVEPQDQStripping = enable_ovep_weight_sharing && !enable_ovep_qdq_optimizer;
778
779
779
780
if (node_unit->UnitType () == NodeUnit::Type::SingleNode) {
780
- AddStandaloneNodeUnit (dst_graph, src_graph, *node_unit, initializers_to_keep, IsWeightSharingWithoutOVEPQDQStripping, logger );
781
+ AddStandaloneNodeUnit (dst_graph, src_graph, *node_unit, initializers_to_keep, logger, IsWeightSharingWithoutOVEPQDQStripping );
781
782
} else {
782
- AddQDQNodeUnit (dst_graph, src_graph, *node_unit, initializers_to_keep, IsWeightSharingWithoutOVEPQDQStripping, logger );
783
+ AddQDQNodeUnit (dst_graph, src_graph, *node_unit, initializers_to_keep, logger, IsWeightSharingWithoutOVEPQDQStripping );
783
784
}
784
785
785
786
seen_node_units.insert (node_unit);
0 commit comments