pytorch/caffe2/python/ideep
Yinghai Lu a839a67aad Add IDEEP unit test with zero-dim tensors (#8459)
Summary:
This test flushes out the issue that IDEEP cannot handle tensor with dims like (0, 2), which is a valid tensor shape.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/8459

Differential Revision: D10419328

Pulled By: yinghai

fbshipit-source-id: c5efcd152364a544180a8305c47a2a2d126ab070
2018-10-19 23:57:33 -07:00
..
concat_split_op_test.py Fix fbcode compatibility (#7939) 2018-05-30 13:35:46 -04:00
conv_op_test.py Enable Conv fusion optimizations in optimizeForIdeep (#9255) 2018-07-16 21:28:50 -07:00
convfusion_op_test.py Fold AffineChannel to Conv, the same way as BN (for Detectron models) (#10293) 2018-08-13 22:43:37 -07:00
copy_op_test.py Add IDEEP unit test with zero-dim tensors (#8459) 2018-10-19 23:57:33 -07:00
dropout_op_test.py
elementwise_sum_op_test.py
fc_op_test.py Fix fbcode compatibility (#7939) 2018-05-30 13:35:46 -04:00
LRN_op_test.py
moment_sgd_op_test.py Implementation MomentumSGD/MomentumSGDUpdate operators for mkl-dnn (#11686) 2018-09-27 13:39:59 -07:00
operator_fallback_op_test.py Enable Detectron model inference for CPU and MKL-DNN paths (#10157) 2018-08-29 15:11:01 -07:00
pool_op_test.py
relu_op_test.py
softmax_op_test.py
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