Names map and input are exchanged mistakenly. By sense of Preconditions paragraph they have to be exchanged I suppose, because there is no problem when map and result coincide (in current context).

Is your feature request related to a problem? Please describe.
While porting some code from SKL to cuML, I have noticed the following:

SKL:
from sklearn.model_selection import train_test_split
cuML:
from cuml.preprocessing.model_selection import train_test_split

If I try to do from cuml.model_selection import train_test_split, the following error is displayed:
`ModuleNotFoundE

Segmented reduce uses the same template type OffsetIteratorT for begin and end offsets

static CUB_RUNTIME_FUNCTION cudaError_t cub::DeviceSegmentedReduce::Sum
    (   void *              d_temp_storage,
        size_t &            temp_storage_bytes,
        InputIteratorT      d_in,
        OutputIteratorT     d_out,
        int                 num_segments,
        OffsetIterato