saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/caffe2/python/ideep
History
Cheng,Penghui 926e718d5f Add/fallback some operators for mkl-dnn (#11696) 
Summary:
Implementation LeakyRelu operator for mkl-dnn,the speed-up of a single operation is up to 10X on BDW.
Implementation rashape operator for mkl-dnn,it will resolve occasionally crash issue which use fallback reshape operator.
Implementation CreateBlobQueue and SafeEnqueueBlobs operators,it will resolve crash issue which use fallback operators.
Fallback CreateBlobsQueueDBOp,TensorProtosDBInput,CloseBlobsQueue operators.
Implement adam operator for mkl-dnn,the speed-up of a single operator is up to 6X on BDW.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/11696

Reviewed By: yinghai

Differential Revision: D10100438

Pulled By: wesolwsk

fbshipit-source-id: 0b6e06897cc11e0a8e349d80a870b1e72e47f10d
2019-01-11 12:53:06 -08:00
..
__init__.py Add __init__.py so files get picked up on install (#14898) 2018-12-07 13:40:23 -08:00
adam_op_test.py Add/fallback some operators for mkl-dnn (#11696) 2019-01-11 12:53:06 -08:00
blobs_queue_db_test.py Add/fallback some operators for mkl-dnn (#11696) 2019-01-11 12:53:06 -08:00
channel_shuffle_op_test.py Implementation of ChannelShuffle Op for MKLDNN (#15106) 2018-12-12 20:25:12 -08:00
concat_split_op_test.py Fallback to CPU concat op to handle TensorCPU inputs (#15263) 2019-01-07 11:13:23 -08:00
conv_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
convfusion_op_test.py Enable conv+add fusion, same as conv+sum (#15268) 2019-01-07 14:42:45 -08:00
copy_op_test.py support 0 size in any of the tensor dimensions in mkldnn (#15295) 2019-01-04 22:33:18 -08:00
dropout_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
elementwise_sum_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
fc_op_test.py Implementation of WeightedSum op for mkl-dnn and fix FC op output shape issue. 2018-12-07 12:35:19 -08:00
leaky_relu_op_test.py Add/fallback some operators for mkl-dnn (#11696) 2019-01-11 12:53:06 -08:00
LRN_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
moment_sgd_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
operator_fallback_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
pool_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
relu_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
reshape_ops_test.py Add/fallback some operators for mkl-dnn (#11696) 2019-01-11 12:53:06 -08:00
sigmoid_op_test.py Add sigmoid op based on MKL-DNN 2018-11-19 22:56:35 -08:00
softmax_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
spatial_bn_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
squeeze_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
test_ideep_net.py
transform_ideep_net.py
weightedsum_op_test.py Implementation of WeightedSum op for mkl-dnn and fix FC op output shape issue. 2018-12-07 12:35:19 -08:00