Skip to content

Commit

Permalink
Update onnxruntime/core/optimizer/transpose_optimization/onnx_transpo…
Browse files Browse the repository at this point in the history
…se_optimization.cc

Co-authored-by: Scott McKay <[email protected]>
  • Loading branch information
adrianlizarraga and skottmckay authored Sep 24, 2024
1 parent 61026ee commit f8cb77a
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -2826,7 +2826,7 @@ static bool TryRemoveEmptyDQQ(OptimizerCtx& ctx, api::NodeRef& q_node) {
return false;
}

// Move Q's output to come out of DQ's parent node instead.
// Move Q's output to come out of DQ's parent node so the graph output value name is maintained.
dq_node.SetInput(0, ""); // Disconnect DQ from its parent first.
ctx.graph.MoveOutput(q_node, 0, *dq_parent_node, dq_parent_output_index);

Expand Down

0 comments on commit f8cb77a

Please sign in to comment.