Both in_data[syncbatchnorm::kData] and out_data[syncbatchnorm::kOut] are TBlob, you only need to write a common function to process both of them so only one TBlob argument 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]
