pytorch
f866e8b5 - [fx2trt] Add trt splitter setting (#71717)

Commit
2 years ago
[fx2trt] Add trt splitter setting (#71717) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71717 Add a setting class for trt splitter which has a specific setting `use_implicit_batch_dim`. Further diffs will try to merge `fx2trt/split.py` and trt splitter. Test Plan: CI Reviewed By: wushirong Differential Revision: D33745251 fbshipit-source-id: 5192da9c9b69d86839a8f26636852d405a40cfe7 (cherry picked from commit e2b0ccb1fab82eb54145404f7fce82294693f9a4)
Author
Committer
Parents
Loading