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 12ac717 commit be0470cCopy full SHA for be0470c
modelopt/onnx/quantization/graph_utils.py
@@ -170,7 +170,7 @@ def has_path_type(
170
def get_fusible_backbone(node: Node, graph: Graph) -> Node | None:
171
"""Returns the linear backbone node for a given node if it matches the pattern.
172
173
- 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.
174
This rule tries to match some of those patterns.
175
Note. BiasAdd and ConstMul are optional in path types.
176
0 commit comments