Replace the use of the input type by the output type when
determining memory type. This allows to handle test graphs with
no variable input.

Note that the memory buffer should actually not depend
on a single output type and be of char type,
this is left as a future change.