Actually, out_data always has same memory description with fwd->fwd_pd, so we should use GetMKLDNNData instead of GetMKLDNNDataReorder. Then out_mem will always be the memory in out_data.
[ Full content available at: https://github.com/apache/incubator-mxnet/pull/12530 ] This message was relayed via gitbox.apache.org for [email protected]
