frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[quant][graphmode] Add Finalize function that inlines graph and produce quantized ops (#33927)

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/33927

Test Plan:
test will be added in later PRs

Imported from OSS

Differential Revision: D20354879

fbshipit-source-id: 03976f4b86c46dbdc4e45764a1e72f1a3855a404
This commit is contained in:
Jerry Zhang
2020-03-12 14:46:25 -07:00
committed by Facebook GitHub Bot
parent 9f05fc9322
commit 90ca7a1feb
7 changed files with 62 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torch/csrc/jit/python/init.cpp
View File

@ -222,6 +222,7 @@ void initJITBindings(PyObject* module) {
[](script::Module& module) {
SwapFunctionalLinear(module);
})
.def("_jit_pass_quant_finalize", &Finalize)
.def(
"_jit_pass_pattern_based_rewrite",
[](const Module& m) { return PatternBasedRewrite(m); })