Viswanath Sivakumar 231d6f7b09 Add SqueezeOp in MKLDNN ...

Summary:
SqueezeOp support to drop drop dims of size 1. MKLMemory now supports Reshape()
if the buffer is in plain layout, in which case just the dims and layouts are
modified similar to caffe2::Tensor. SqueezeOp takes care of converting the
input to plain layout if needed via an intermediate buffer before calling
Reshape().

Differential Revision: D6735656

fbshipit-source-id: 953309498370e1b8986e8c593bc6963f38036255

2018-01-22 18:39:42 -08:00

..

convnet_benchmarks.py

Re-license to Apache

2017-09-28 16:22:00 -07:00

mkl_concat_op_test.py

Add MKLConcatOp

2018-01-11 14:19:22 -08:00

mkl_conv_op_test.py

Re-license to Apache

2017-09-28 16:22:00 -07:00

mkl_copy_op_test.py

Fix crash due to copying empty tensors into MKLMemory

2017-12-30 15:36:48 -08:00

mkl_elementwise_add_op_test.py

Add op in MKLDNN

2018-01-21 08:21:43 -08:00

mkl_elementwise_sum_op_test.py

Re-license to Apache

2017-09-28 16:22:00 -07:00

mkl_fc_op_test.py

Re-license to Apache

2017-09-28 16:22:00 -07:00

mkl_fc_speed_test.py

Re-license to Apache

2017-09-28 16:22:00 -07:00

mkl_fill_op_test.py

Re-license to Apache

2017-09-28 16:22:00 -07:00

mkl_LRN_op_test.py

Re-license to Apache

2017-09-28 16:22:00 -07:00

mkl_LRN_speed_test.py

Re-license to Apache

2017-09-28 16:22:00 -07:00

mkl_pool_op_test.py

Re-license to Apache

2017-09-28 16:22:00 -07:00

mkl_pool_speed_test.py

Re-license to Apache

2017-09-28 16:22:00 -07:00

mkl_relu_op_test.py

Re-license to Apache

2017-09-28 16:22:00 -07:00

mkl_sbn_op_test.py

disable sbn running mean and var comp

2017-11-13 12:09:08 -08:00

mkl_sbn_speed_test.py

Re-license to Apache

2017-09-28 16:22:00 -07:00

mkl_sigmoid_op_test.py

MKL Sigmoid op wrapper

2017-11-02 17:30:29 -07:00

mkl_speed_test.py

Re-license to Apache

2017-09-28 16:22:00 -07:00

mkl_squeeze_op_test.py

Add SqueezeOp in MKLDNN

2018-01-22 18:39:42 -08:00

rewrite_graph_test.py

More extensions

2017-11-20 17:18:51 -08:00

rewrite_graph.py

Updates to MKL conversion script

2018-01-08 12:25:23 -08:00