Skip to content

Commit a96b108

Browse files
committed
nit: added MaxPool to comment
Signed-off-by: gcunhase <[email protected]>
1 parent 49d7c38 commit a96b108

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
@@ -173,7 +173,7 @@ def has_path_type(
173173
def get_fusible_backbone(node: Node, graph: Graph) -> Node | None:
174174
"""Returns the linear backbone node for a given node if it matches the pattern.
175175
176-
TensorRT fuses convolution with BN, Relu etc. when in some specific pattern.
176+
TensorRT fuses convolution with BN, Relu, MaxPool etc. when in some specific pattern.
177177
This rule tries to match some of those patterns.
178178
Note. BiasAdd and ConstMul are optional in path types.
179179

0 commit comments

Comments
 (0)