pytorch
9cef2033 - Modify decorator for acc op converters (#67636)

Commit
4 years ago
Modify decorator for acc op converters (#67636) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/67636 Modify decorator to denote whether a acc op converter is able to support explicit/implicit batch dim. This info will be used by trt_splitter when determine whether a node can be split into acc graph. This is can prevent us from split node to acc module and later found no proper converter exist for the node and fail the lower process. Test Plan: unit test Reviewed By: 842974287 Differential Revision: D31998477 fbshipit-source-id: 6789ebef4a76f9a0c1ab3edf8e846a5b6143326b
Author
Shirong Wu
Parents
Loading