Cheng,Penghui
e13101e069
support pre-convert filter format for mkldnn training mode and change 'OptimizeForIdeep' to 'OptimizeForMkldnn' ( #15171 )
...
Summary:
For MKL-DNN,the filter data will be reorderd to primitive format, it takes a lot of time.
So the patch provide a method to convert filter format before training.
And "OptimizeForIdeep" will be changed to "OptimizeForMkldnn" in this patch.
This patch depends on https://github.com/pytorch/pytorch/pull/12866
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15171
Differential Revision: D14590741
Pulled By: yinghai
fbshipit-source-id: 07971c9977edac3c8eec08ca2c39cda639683492
2019-03-29 19:00:48 -07:00
Duc Ngo
66556f48e3
Remove sinkMaxPool transformation ( #17694 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17694
Remove sinkMaxPool transformation as it's unused
Differential Revision: D14328624
fbshipit-source-id: bd245403b756157120faa61a0f9253c15120e7a8
2019-03-12 20:10:46 -07:00
Gu, Jinghui
e8b8c3895e
Enable Conv fusion optimizations in optimizeForIdeep ( #9255 )
...
Summary:
Enable fusion for IDEEP in optimizeForIdeep
including Conv+ReLU, Conv+Sum, Conv+Sum+ReLU, Conv+BN
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9255
Reviewed By: bddppq
Differential Revision: D8809030
Pulled By: yinghai
fbshipit-source-id: af30bad3b96cb965bd26a4dfa810370faec4bb88
2018-07-16 21:28:50 -07:00
Orion Reblitz-Richardson
9ec0a2aef4
fbshipit-source-id: ba600fcd2b5cefc7621357bdeb05e24cea02e5af
2018-06-27 04:50:56 -07:00
Bram Wasti
3913e9ead3
[caffe2][nomnigraph] Batchnorm + Conv Fusion ( #7057 )
2018-05-08 15:40:34 -07:00
Yinghai Lu
e3935f7509
[Caffe2] Add conv+relu fusion for MKLDNN ops (IDEEP) ( #7385 )
...
* Add conv+relu fusion for MKLDNN ops (IDEEP)
* comments
2018-05-08 14:44:53 -07:00
Bram Wasti
3642745ef9
[caffe2][nomnigraph] Add maxpool sink transform ( #7207 )
2018-05-07 14:52:10 -07:00
Bram Wasti
7bd398b3db
Add fuseNNPACKConvRelu ( #6439 )
2018-04-10 16:51:16 -07:00
Bram Wasti
ee64200c64
[nomnigraph] Expose transformations to python
...
Adding a python interface to the transformations
2018-03-30 21:00:44 -07:00
