NN split and early exit