Join GitHub today
GitHub is home to over 40 million developers working together to host and review code, manage projects, and build software together.Sign up
[v1.5.x] Handle fix_gamma in tensorrt subgraph conversion correctly #15874
the current nnvm->onnx->tensorrt conversion is ignoring the fix_gamma field for batch norm layer. This commit tries to solve this particular issue by providing a way to massage params map before convert nnvm to onnx.
Please feel free to remove inapplicable items for your PR.
This is a bug fix for tensor rt subgraph conversion