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
conv_op_test.py
convfusion_op_test.py
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
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
pool_op_test.py
relu_op_test.py
softmax_op_test.py
spatial_bn_op_test.py
squeeze_op_test.py
test_ideep_net.py
transform_ideep_net.py