OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
7103d0d938
pytorch/caffe2/python/ideep
History
Cheng,Penghui 6e7e63fda3 Implementation MomentumSGD/MomentumSGDUpdate operators for mkl-dnn (#11686) 
Summary:
the speed-up of a single operation is up to 6X on BDW.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/11686

Reviewed By: yinghai

Differential Revision: D9828129

Pulled By: wesolwsk

fbshipit-source-id: 7dbacea90609e18438f6fe1229c641937d0696c8
2018-09-27 13:39:59 -07:00
..
concat_split_op_test.py Fix fbcode compatibility (#7939) 2018-05-30 13:35:46 -04:00
conv_op_test.py Enable Conv fusion optimizations in optimizeForIdeep (#9255) 2018-07-16 21:28:50 -07:00
convfusion_op_test.py Fold AffineChannel to Conv, the same way as BN (for Detectron models) (#10293) 2018-08-13 22:43:37 -07:00
copy_op_test.py Add fallback to TensorCPU if there are unsupported types for IDEEP Tensor (#9667) 2018-07-23 13:54:57 -07:00
dropout_op_test.py [feature request] [Caffe2] Enable MKLDNN support for inference (#6699) 2018-04-22 21:58:14 -07:00
elementwise_sum_op_test.py [feature request] [Caffe2] Enable MKLDNN support for inference (#6699) 2018-04-22 21:58:14 -07:00
fc_op_test.py Fix fbcode compatibility (#7939) 2018-05-30 13:35:46 -04:00
LRN_op_test.py [Caffe2] [feature request] Add gradient operators for IDEEP (#7234) 2018-05-09 08:52:24 -07:00
moment_sgd_op_test.py Implementation MomentumSGD/MomentumSGDUpdate operators for mkl-dnn (#11686) 2018-09-27 13:39:59 -07:00
operator_fallback_op_test.py Enable Detectron model inference for CPU and MKL-DNN paths (#10157) 2018-08-29 15:11:01 -07:00
pool_op_test.py [Caffe2] [feature request] Add gradient operators for IDEEP (#7234) 2018-05-09 08:52:24 -07:00
relu_op_test.py [Caffe2] [feature request] Add gradient operators for IDEEP (#7234) 2018-05-09 08:52:24 -07:00
softmax_op_test.py [feature request] [Caffe2] Enable MKLDNN support for inference (#6699) 2018-04-22 21:58:14 -07:00
spatial_bn_op_test.py Normalize IDEEP spatial bn op test (#9276) 2018-07-09 11:55:41 -07:00
squeeze_op_test.py [ideep] Add IDEEP Squeeze op (#8227) 2018-06-06 21:58:51 -07:00
test_ideep_net.py [Caffe2] Fix of the performance issue of IDEEP (#7503) 2018-05-11 13:43:41 -07:00
transform_ideep_net.py [Caffe2] Ideep net optmization passes (#7514) 2018-05-11 23:50:18 -07:00