..
__init__.py
Add __init__.py so files get picked up on install ( #14898 )
2018-12-07 13:40:23 -08:00
adam_op_test.py
Add/fallback some operators for mkl-dnn ( #11696 )
2019-01-11 12:53:06 -08:00
blobs_queue_db_test.py
Add/fallback some operators for mkl-dnn ( #11696 )
2019-01-11 12:53:06 -08:00
channel_shuffle_op_test.py
Implementation of ChannelShuffle Op for MKLDNN ( #15106 )
2018-12-12 20:25:12 -08:00
concat_split_op_test.py
support negative axis in concat and split operators
2019-06-10 15:26:29 -07:00
conv_op_test.py
caffe2 python ideep conv_op test_int8_convolution skip for python 3
2019-10-08 21:31:11 -07:00
conv_transpose_test.py
Removed work around for convolution transpose op since the bug has be… ( #22184 )
2019-06-25 14:34:34 -07:00
convfusion_op_test.py
Improve optimizations for DNNLOWP support on MKL-DNN ( #18843 )
2019-04-20 02:12:06 -07:00
copy_op_test.py
support 0 size in any of the tensor dimensions in mkldnn ( #15295 )
2019-01-04 22:33:18 -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
implement operators for DNNLOWP ( #18656 )
2019-04-10 12:04:39 -07:00
expanddims_squeeze_op_test.py
Impl ExpandDims op and fallback to CPU if needed ( #15264 )
2019-02-08 12:04:53 -08:00
fc_op_test.py
implement operators for DNNLOWP ( #18656 )
2019-04-10 12:04:39 -07:00
leaky_relu_op_test.py
Add/fallback some operators for mkl-dnn ( #11696 )
2019-01-11 12:53:06 -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
order_switch_op_test.py
implement operators for DNNLOWP ( #18656 )
2019-04-10 12:04:39 -07:00
pool_op_test.py
implement operators for DNNLOWP ( #18656 )
2019-04-10 12:04:39 -07:00
pre_convert_test.py
support pre-convert filter format for mkldnn training mode and change 'OptimizeForIdeep' to 'OptimizeForMkldnn' ( #15171 )
2019-03-29 19:00:48 -07:00
relu_op_test.py
implement operators for DNNLOWP ( #18656 )
2019-04-10 12:04:39 -07:00
reshape_op_test.py
Support int64_t shape data for ideep reshape op
2019-01-30 09:00:09 -08:00
shape_op_test.py
support pre-convert filter format for mkldnn training mode and change 'OptimizeForIdeep' to 'OptimizeForMkldnn' ( #15171 )
2019-03-29 19:00:48 -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
test_ideep_net.py
transform_ideep_net.py
support pre-convert filter format for mkldnn training mode and change 'OptimizeForIdeep' to 'OptimizeForMkldnn' ( #15171 )
2019-03-29 19:00:48 -07:00
transpose_op_test.py
implement transpose operator for MKLDNN ( #19955 )
2019-06-11 01:55:13 -07:00
weightedsum_op_test.py
Implementation of WeightedSum op for mkl-dnn and fix FC op output shape issue.
2018-12-07 12:35:19 -08:00