pytorch
e4627e5d - [quant][graphmode] Fix add_relu patterns for scripting and tracing (#39455)

Commit
4 years ago
[quant][graphmode] Fix add_relu patterns for scripting and tracing (#39455) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/39455 1. enable filters in PatternInfo 2. add aten_add_alpha_is_one filter 3. add is_functional_relu filter 4. add is_relu_module filter 5. fix the relu module method call matching in traced modules with regex 6. add aten::add - aten::relu patterns for traced modules Test Plan: Imported from OSS Differential Revision: D21917118 fbshipit-source-id: e67b55cd1c070fd4238f563d933a6f10a3582ae3
Author
Parents
Loading