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
Gu, Jinghui 60963c2ecb Add "axis" and "axis_w" arguments in FC to support customized axix to reduce dim. (#12971) 
Summary:
Add "axis" and "axis_w" arguments in FC to support customized axix to reduce dim.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12971

Reviewed By: bddppq

Differential Revision: D12850675

Pulled By: yinghai

fbshipit-source-id: f1cde163201bd7add53b8475329db1f038a73019
2018-11-21 15:44:50 -08:00
..
concat_split_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
conv_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
convfusion_op_test.py seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
copy_op_test.py build with mkl-dnn by default (#13303) 2018-11-08 11:18:27 -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 Add "axis" and "axis_w" arguments in FC to support customized axix to reduce dim. (#12971) 2018-11-21 15:44:50 -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
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