[GitHub] ptrendx commented on issue #13362: Add NHWC layout support to Pooling (cuDNN only)

[GitBox <gi...@apache.org>]

ptrendx commented on issue #13362: Add NHWC layout support to Pooling (cuDNN only)
URL: https://github.com/apache/incubator-mxnet/pull/13362#issuecomment-444669921
 
 
   @TaoLv 
   Re MKLDNN: Per title of this PR this is enabling NHWC support only for cuDNN path. I don't know whether MKLDNN supports NHWC pooling.
   Re quantized pooling: that file seems to assume NCHW layout. What would you like me to do with it?
   Re layout parameter: Yes, other operators expose layout information (in fact, pooling already does as well) - see convolutions (e.g. here https://mxnet.incubator.apache.org/api/python/gluon/nn.html#mxnet.gluon.nn.Conv2D) and batchnorm (axis parameter). NDArray does not know its layout and for many use cases it would not make sense. That is why layout is exposed as an operator parameter.

----------------------------------------------------------------
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:
users@infra.apache.org


With regards,
Apache Git Services