Add arguments `(in|out)put_segments`

Add arguments `input_segments` and `output_segments`. Because on
one occasion I wanted to use and keep two versions of input and output
graphs in the same directory