We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 49d7c38 commit a96b108Copy full SHA for a96b108
modelopt/onnx/quantization/graph_utils.py
@@ -173,7 +173,7 @@ def has_path_type(
173
def get_fusible_backbone(node: Node, graph: Graph) -> Node | None:
174
"""Returns the linear backbone node for a given node if it matches the pattern.
175
176
- TensorRT fuses convolution with BN, Relu etc. when in some specific pattern.
+ TensorRT fuses convolution with BN, Relu, MaxPool etc. when in some specific pattern.
177
This rule tries to match some of those patterns.
178
Note. BiasAdd and ConstMul are optional in path types.
179
0 commit comments