Enabled conv2d batch norm fusion for traced modules.
Summary:
This pass enabled fusion of conv2d with batch norm for traced modules.
It requires that the module be preprocessed with freeze_module pass,
which inlines the graph constant propgates all attributes.
This pass work on top of such a graph.
However, at the moment it does not work with scripted module.
For some reason subgraph matcher is not able to find the fusion pattern,
when it operates on scripted module.
Test Plan:
python test/test_jit.py TestJit.test_foldbn_graph_mode
Reviewers:
Subscribers:
Tasks:
Tags: