DickJC123 commented on issue #13362: Add NHWC layout support to Pooling (cuDNN only) URL: https://github.com/apache/incubator-mxnet/pull/13362#issuecomment-444734362 The forced transposition I referred to is done currently by the user in the python model. With this new feature, the Transpose operators could get removed in favor of a Pooling with the new layout parameter. I'm a bit sorry that NDArray's don't carry the layout information. As a result, the layout has to travel along with the data ('out-of-band') and be passed into the operators along the way. On the other hand, having multi-input operators deal with inconsistent layouts would be a headache. Anyway, that's a discussion to be debated in its own thread.
---------------------------------------------------------------- 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: us...@infra.apache.org With regards, Apache Git Services