[AutoDiff] Canonicalize JVP/VJP types to re-enable LoadableByAddress.
Canonicalize JVP/VJP types so the returned original result is always indirect.
This makes it so JVP/VJP type no longer depends on the abstraction of the
original function type, making it possible to re-enable LoadableByAddress.
Resolves TF-626.