pytorch
fccaa4a3 - [fx2trt] fix transpose unittest (#65481)

Commit
3 years ago
[fx2trt] fix transpose unittest (#65481) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65481 Previous we have `acc_ops.transpose` but after a recent diff `torch.transpose` is mapped to `acc_ops.permute`. Here we clean up the fx2trt unittest for transpose and add support for negative indices in permute. Reviewed By: wushirong Differential Revision: D31115280 fbshipit-source-id: 58e689e6dd14181aea5186f3bb5b8745a07d0e51
Author
Parents
Loading