Features: update the fx model for output dtype match #2040

wu6u3tw wants to merge 14 commits intopytorch:mainpytorch/TensorRT:mainfrom

wu6u3tw:dev_tinyinl_nvbugwu6u3tw/TensorRT:dev_tinyinl_nvbugCopy head branch name to clipboard