[GitHub] sandeep-krishnamurthy opened a new issue #11192: BatchNorm operator on GPUs are slow with channels_last

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

sandeep-krishnamurthy opened a new issue #11192: BatchNorm operator on GPUs are slow with channels_last
URL: https://github.com/apache/incubator-mxnet/issues/11192
 
 
   [mx.sym.BatchNorm operator](https://mxnet.incubator.apache.org/api/python/symbol/symbol.html#mxnet.symbol.BatchNorm) is considerably slower on GPUs with channels_last (axis=-1).
   
   My understanding is that, with channels_first (axis=1) operator calls cuDNN implementation and with channels_last(axis=-1) MXNet has its own implementation.
   
   

----------------------------------------------------------------
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