Add typing annotations in doc_controls to preserve decorated function typing
Add typing annotations in doc_controls to preserve decorated function typing

PiperOrigin-RevId: 339268789

tensorflow/docs,tensorflow/docs,tensorflow/docs