def METHOD_NAME(self, gc, dc):
    input_lengths = np.array([9, 4, 5]).astype(np.int32)
    n_split = 3
    # Expected output:
    # [3, 3, 3, 2, 1, 1, 2, 2, 1]
    op = core.CreateOperator(
        'LengthsSplit',
        ['input_lengths'],
        ['Y'], n_split=n_split
    )
    # Check against numpy reference
    self.assertReferenceChecks(
        device_option=gc,
        op=op,
        inputs=[input_lengths],
        reference=lambda x : self._length_split_op_ref(x, [n_split]),
    )
    # Check over multiple devices
    self.assertDeviceChecks(dc, op, [input_lengths], [0])