minimized when the `Split` is with its predecessor. This colocation
    constraint will be used by the placer graph optimization to assign a device
    to the op.

    This pass should run in the export pipeline after tf-replicate-to-island so
    each replica has its own distinct (predecessor, Split) pair.

    The colocation class (`_class`) of the `Split` is set to the same class as
    its predecessor: