https://github.com/NVIDIA/TensorRT-Model-Optimizer/blob/4c611e47a60084a86e1de7e48690a692a1b8170c/modelopt/torch/_deploy/utils/torch_onnx.py#L560