frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-02 06:24:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

jit: Conv3d + BatchNorm3d fusion (#40082)

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

Differential Revision: D22120340

Pulled By: jerryzh168

fbshipit-source-id: fce6c5f03fe7ab6c60620cbdf547d5a466a470e3
This commit is contained in:
Zhang, Xiaobing
2020-06-22 11:14:18 -07:00
committed by Facebook GitHub Bot
parent 14f7e95c1a
commit 87c5f02f3d
6 changed files with 130 additions and 173 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -230,7 +230,7 @@ void initJITBindings(PyObject* module) {
.def(
"_jit_pass_quant_fusion",
[](std::shared_ptr<Graph>& g) { return QuantFusion(g); })
.def("_jit_pass_fold_convbn", &FoldConvBatchNorm2d)
.def("_jit_pass_fold_convbn", &FoldConvBatchNorm)
.def(
"_freeze_module",
[](Module& module, std::vector<std::string>& preservedAttrs) {