Simplify lines by removing the input_size function and use offset instead.

This MR simplifies some lines by removing the need to define input_size, reusing offset for that purpose instead.