A distributed approach for millimetre-wave electron device modelling