pytorch/caffe2/python/mkl
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
mkl_concat_op_test.py Add MKLConcatOp 2018-01-11 14:19:22 -08:00
mkl_conv_op_test.py
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
mkl_fc_op_test.py
mkl_fc_speed_test.py
mkl_fill_op_test.py
mkl_LRN_op_test.py
mkl_LRN_speed_test.py
mkl_pool_op_test.py
mkl_pool_speed_test.py
mkl_relu_op_test.py
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
mkl_sigmoid_op_test.py MKL Sigmoid op wrapper 2017-11-02 17:30:29 -07:00
mkl_speed_test.py
mkl_squeeze_op_test.py Add SqueezeOp in MKLDNN 2018-01-22 18:39:42 -08:00
rewrite_graph.py Updates to MKL conversion script 2018-01-08 12:25:23 -08:00
rewrite_graph_test.py More extensions 2017-11-20 17:18:51 -08:00