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 8143a8f commit 5aa52f6Copy full SHA for 5aa52f6
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