OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
as_tensor_docs
pytorch/caffe2/ideep
History
Yury Gitman 9c8f5cb61d Ensure IDEEP transpose operator works correctly 
Summary: I found out that without exporting to public format IDEEP transpose operator in the middle of convolution net produces incorrect results (probably reading some out-of-bound memory). Exporting to public format might not be the most efficient solution, but at least it ensures correct behavior.

Test Plan: Running ConvFusion followed by transpose should give identical results on CPU and IDEEP

Reviewed By: bwasti

Differential Revision: D22970872

fbshipit-source-id: 1ddca16233e3d7d35a367c93e72d70632d28e1ef
2020-08-11 12:58:31 -07:00
..
operators Ensure IDEEP transpose operator works correctly 2020-08-11 12:58:31 -07:00
utils Refactor error msg stack handling, add TORCH_RETHROW (#37101) 2020-05-04 11:56:45 -07:00
CMakeLists.txt seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
ideep_utils.h Upgrade MKL-DNN to DNNL v1.2 (#32422) 2020-03-26 22:07:59 -07:00