Skip to content

Commit

Permalink
Update comment
Browse files Browse the repository at this point in the history
  • Loading branch information
adrianlizarraga committed Jul 22, 2024
1 parent f4390e7 commit 51ace3a
Showing 1 changed file with 13 additions and 12 deletions.
25 changes: 13 additions & 12 deletions onnxruntime/core/optimizer/qdq_transformer/qdq_propagation.cc
Original file line number Diff line number Diff line change
Expand Up @@ -118,19 +118,20 @@ void LogQDQInsertion(const logging::Logger& logger, logging::Severity severity,
<< " at NodeArg \"" << node_arg_name << "\".";
}

// convert this: src_node --+--> dst_node_0
// |
// +--> dst_node_1
// | ...
// +--> dst_node_n
// convert this: src_node (or graph input) --+--> dst_node_0 (or graph output)
// |
// +--> dst_node_1
// | ...
// +--> dst_node_n
//
// to this: src_node -> Q --+--> DQ -> dst_node_0
// |
// +--> DQ -> dst_node_1
// | ...
// +--> DQ -> dst_node_n
// Checks that all insertion edges share the same NodeArg. That is, the edges have the same source node and the
// same source node output index. This function returns an error status if edges are invalid.
// to this: src_node (or graph input) -> Q --+--> DQ -> dst_node_0 (or graph output)
// |
// +--> DQ -> dst_node_1
// | ...
// +--> DQ -> dst_node_n
// Checks that all insertion edges share the same NodeArg. That is, the edges originate from the same source node
// output. If there is no src_node, then all edges should come from the same graph input.
// This function returns an error status if edges are invalid.
//
// Assumes that scale_initializer_nodearg and zp_initializer_nodearg_ptr (if not null) are constant initializers.
Status InsertQDQPairs(Graph& graph, gsl::span<const ExtendedGraphEdge> insertion_edges,
Expand Down

0 comments on commit 51ace3a

Please sign in to comment.