Both in_data[syncbatchnorm::kData] and out_data[syncbatchnorm::kOut] are TBlob data type, you only need to write a single common function to process both in_data and out_data. In that case, only one argument TBlob & is needed. As for workspace, I think you can wrap that inside the function as well with some proper refactoring.
[ Full content available at: https://github.com/apache/incubator-mxnet/pull/12261 ] This message was relayed via gitbox.apache.org for [email protected]
