mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
926e718d5f
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 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| adam_op_test.py | ||
| blobs_queue_db_test.py | ||
| channel_shuffle_op_test.py | ||
| concat_split_op_test.py | ||
| conv_op_test.py | ||
| convfusion_op_test.py | ||
| copy_op_test.py | ||
| dropout_op_test.py | ||
| elementwise_sum_op_test.py | ||
| fc_op_test.py | ||
| leaky_relu_op_test.py | ||
| LRN_op_test.py | ||
| moment_sgd_op_test.py | ||
| operator_fallback_op_test.py | ||
| pool_op_test.py | ||
| relu_op_test.py | ||
| reshape_ops_test.py | ||
| sigmoid_op_test.py | ||
| softmax_op_test.py | ||
| spatial_bn_op_test.py | ||
| squeeze_op_test.py | ||
| test_ideep_net.py | ||
| transform_ideep_net.py | ||
| weightedsum_op_test.py | ||