support pre-convert filter format for mkldnn training mode and change 'OptimizeForIde...
authorCheng,Penghui <penghui.cheng@intel.com>
Sat, 30 Mar 2019 01:51:50 +0000 (18:51 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Sat, 30 Mar 2019 02:00:48 +0000 (19:00 -0700)
commite13101e0691b0eabc1900f482a615ea7f14e7a72
treeda2f742418023d682676b1bffa4b46e6e08848db
parentd73c830e236f5b980e5c91914b818d150b60278c
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
12 files changed:
caffe2/ideep/ideep_utils.h
caffe2/opt/converter.cc
caffe2/opt/optimize_ideep.cc
caffe2/opt/optimize_ideep.h
caffe2/python/ideep/conv_op_test.py
caffe2/python/ideep/convfusion_op_test.py
caffe2/python/ideep/pre_convert_test.py [new file with mode: 0644]
caffe2/python/ideep/shape_op_test.py
caffe2/python/ideep/transform_ideep_net.py
caffe2/python/pybind_state.cc
caffe2/python/pybind_state_ideep.cc
caffe2/python/transformations.py