Skip to content

Commit be0470c

Browse files
committed
nit: added MaxPool to comment
Signed-off-by: gcunhase <4861122+gcunhase@users.noreply.github.com>
1 parent 12ac717 commit be0470c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modelopt/onnx/quantization/graph_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ def has_path_type(
170170
def get_fusible_backbone(node: Node, graph: Graph) -> Node | None:
171171
"""Returns the linear backbone node for a given node if it matches the pattern.
172172
173-
TensorRT fuses convolution with BN, Relu etc. when in some specific pattern.
173+
TensorRT fuses convolution with BN, Relu, MaxPool etc. when in some specific pattern.
174174
This rule tries to match some of those patterns.
175175
Note. BiasAdd and ConstMul are optional in path types.
176176

0 commit comments

Comments
 (0)