mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
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 |
||
|---|---|---|
| .. | ||
| convnet_benchmarks.py | ||
| mkl_concat_op_test.py | ||
| mkl_conv_op_test.py | ||
| mkl_copy_op_test.py | ||
| mkl_elementwise_add_op_test.py | ||
| 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 | ||
| mkl_sbn_speed_test.py | ||
| mkl_sigmoid_op_test.py | ||
| mkl_speed_test.py | ||
| mkl_squeeze_op_test.py | ||
| rewrite_graph.py | ||
| rewrite_graph_test.py | ||