Let me add some logging code to see what types are used here. My assumption is that uint64 is used for all size operations. It's large enough to handle large indexes and it's integer type, which makes sense for sizes (as opposed to floats). Let me do a few test runs here in CI and validate those assumptions.
[ Full content available at: https://github.com/apache/incubator-mxnet/pull/12631 ] This message was relayed via gitbox.apache.org for [email protected]
