mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
e13101e069
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 |
||
|---|---|---|
| .. | ||
| __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 | ||
| conv_transpose_test.py | ||
| convfusion_op_test.py | ||
| copy_op_test.py | ||
| dropout_op_test.py | ||
| elementwise_sum_op_test.py | ||
| expanddims_squeeze_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 | ||
| pre_convert_test.py | ||
| relu_op_test.py | ||
| reshape_op_test.py | ||
| shape_op_test.py | ||
| sigmoid_op_test.py | ||
| softmax_op_test.py | ||
| spatial_bn_op_test.py | ||
| test_ideep_net.py | ||
| transform_ideep_net.py | ||
| weightedsum_op_test.py | ||