xinyu-intel commented on issue #13530: Integrate MKLDNN Conv1d and support 3d layout URL: https://github.com/apache/incubator-mxnet/pull/13530#issuecomment-449811865 @pengzhao-intel Conv1d will fall back to native cpu implement. 100 iterations total time(ms) on Xeon Skylake 8180 1 socket: |shape|before opt|after opt|speedup| |:---:|:---:|:---:|:---:| |(1,256,200)|715.47|73.88|9.68x| |(1,1024,512)|1970.15|101.06|19.49x| |(64,1024,512)|131312.48|4196.24|31.29x| I've add 1d,3d,4d data shape to activation test. Regarding quantized conv, it will now return error when the data shape is 3d and users should exclude this layer: ``` CHECK_EQ(param.full_conv_param.conv_param.kernel.ndim(), 2U) << "MKL-DNN only supports quantized conv2d."; ```
---------------------------------------------------------------- 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
