Add function signatures as the first line of the docstring for functions with `@device_override` -- otherwise the signatures are not correctly displayed in the docs