soulitzer 2f615f6313 Improve custom function docs (#60312) ... Summary: - Adds some code examples for `ctx` methods and make requirements of arguments more clear - Type annotations for `save_for_backward`, `mark_dirty`, `mark_non_differentiable`, and `set_materialize_grads` (BC-breaking?) - Refactor `torch.autograd.Function` doc Pull Request resolved: https://github.com/pytorch/pytorch/pull/60312 Reviewed By: VitalyFedyunin Differential Revision: D30314961 Pulled By: soulitzer fbshipit-source-id: a284314b65662e26390417bd2b6b12cd85e68dc8		2021-08-18 11:31:31 -07:00
..
amp_examples.rst
autograd.rst	clarify default value of requires_grad for tensors (#61038)	2021-07-12 12:57:37 -07:00
broadcasting.rst	Fixes docs (#51439)	2021-01-31 22:00:26 -08:00
cpu_threading_runtimes.svg
cpu_threading_torchscript_inference.rst
cpu_threading_torchscript_inference.svg	Lint trailing newlines (#54737)	2021-03-30 13:09:52 -07:00
cuda.rst	Add docstrings for save_on_cpu hooks (#62410)	2021-08-03 17:53:45 -07:00
ddp.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
extending.rst	Improve custom function docs (#60312)	2021-08-18 11:31:31 -07:00
faq.rst	Update faq.rst so OOM section mentions checkpoint (#62709)	2021-08-05 07:40:08 -07:00
gradcheck.rst	Add first draft of gradcheck note (#55966)	2021-04-27 14:33:42 -07:00
hip.rst	Add note on torch.distributed backends on ROCm (#58975)	2021-07-10 03:51:19 -07:00
large_scale_deployments.rst
modules.rst	Note on Modules for 1.8 docs (#51536)	2021-02-04 11:28:11 -08:00
multiprocessing.rst
randomness.rst	add comma to prevent syntax errors (#62492)	2021-08-16 12:27:31 -07:00
serialization.rst	docs: reference links to serialization.html (#54659)	2021-03-29 10:15:07 -07:00
windows.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00