public static int cudnnGetReductionIndicesSize ( cudnnHandle handle , cudnnReduceTensorDescriptor reduceTensorDesc , cudnnTensorDescriptor aDesc , cudnnTensorDescriptor cDesc , long [ ] sizeInBytes ) { return checkResult ( cudnnGetReductionIndicesSizeNative ( handle , reduceTensorDesc , aDesc , cDesc , sizeInBytes ) ) ; }
Helper function to return the minimum size of the index space to be passed to the reduction given the input and output tensors