zheng-da commented on issue #11906: support 1D and 3D arrays in MKLDNN. URL: https://github.com/apache/incubator-mxnet/issues/11906#issuecomment-408485145 I know mkldnn has 3D arrays, but not all mkldnn operators support 3D arrays. I believe the current implementation of mkldnn integration only allows 2D and 4D arrays. For example, mkldnn convolution only supports 2D kernel on 4D arrays. Currently, 1D convolution actually calls the native implementation. But if we add a fake dim to turn 1D conv into 2D conv, we can make substantial speedup. I think this applies to many other operators.
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
