soulitzer c88367442d [forward ad] forbid non-float non-complex tangent and primal ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/78361 Approved by: https://github.com/albanD		2022-05-31 20:58:19 +00:00
..
_functions
__init__.py	Monkey patch Variable module to fix FX codegen	2022-04-21 23:07:51 +00:00
anomaly_mode.py	Add __all__ for torch.autograd.{anomaly_mode, gradcheck, forward_ad}	2022-05-10 17:36:47 +00:00
forward_ad.py	[forward ad] forbid non-float non-complex tangent and primal	2022-05-31 20:58:19 +00:00
function.py	Add save_for_forward to custom function (#71569)	2022-01-25 07:30:46 +00:00
functional.py	[lint] upgrade mypy to latest version	2022-05-03 20:51:34 +00:00
grad_mode.py	Add type hints for a few random functions/classes	2022-05-04 13:53:00 +00:00
gradcheck.py	Fix gradcheck when outputs that don't require grad precede those that do	2022-05-24 22:41:49 +00:00
graph.py	Enable nested default hooks (#70932)	2022-01-11 15:03:49 -08:00
profiler_legacy.py	[pytorch][cupti profiler 6/n] Changes to configure Kineto cupti profiler from pytorch profiler interface (#75616)	2022-04-20 22:19:54 +00:00
profiler_util.py	record_function: add torchbind alternative API (#72301)	2022-03-08 03:26:27 +00:00
profiler.py	Back out "record_function: update to use custom_class API" (#76253)	2022-04-26 17:49:57 -04:00
variable.py	[WIP] wrap XLATensors into Python XLA wrapper class (#65841)	2021-10-25 16:11:03 -07:00