pytorch/caffe2/python/ideep
Yinghai Lu cb98c5020a Normalize IDEEP spatial bn op test (#9276)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9276

Use `checkDevice` instead rolling our own.

Reviewed By: orionr

Differential Revision: D8769401

fbshipit-source-id: bd47ec2b2501552c2da1cee2eb9ad96a215602b4
2018-07-09 11:55:41 -07:00
..
concat_split_op_test.py Fix fbcode compatibility (#7939) 2018-05-30 13:35:46 -04:00
conv_op_test.py Add depthwise convolution test for IDEEP (#8301) 2018-06-09 08:44:13 -07:00
convfusion_op_test.py Update from facebook (#7855) 2018-05-29 11:38:02 -07:00
copy_op_test.py cpu/ideep context converter (#8139) 2018-06-04 21:28:59 -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
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