Improve layer docstrings by 1) adding an "Extra call arguments" section...
Improve layer docstrings by 1) adding an "Extra call arguments" section describing `call` arguments such as `training`, `mask`, etc., 2) fixing various formatting issues. PiperOrigin-RevId: 236002535
Loading
Please sign in to comment