pytorch/tools/codegen/api
Michael Dagitses 543185a0fd support using gradients named for outputs in derivatives (#63947)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63947

Fixes #62196

Test Plan: Imported from OSS

Reviewed By: albanD

Differential Revision: D30541485

Pulled By: dagitses

fbshipit-source-id: ea1dd0edd1a51936a295631e52b85e9c022a9c87
2021-09-18 07:31:45 -07:00
..
__init__.py
autograd.py support using gradients named for outputs in derivatives (#63947) 2021-09-18 07:31:45 -07:00
cpp.py Reland of "D27708346: generate xla codegen in-tree" (#56601) 2021-04-21 19:36:31 -07:00
dispatcher.py Reland of "D27708346: generate xla codegen in-tree" (#56601) 2021-04-21 19:36:31 -07:00
meta.py Un-ignore F403 in .flake8 (#55838) 2021-04-13 09:24:07 -07:00
native.py [PyTorch] Fix const correctness for resize native functions (#55351) 2021-04-21 14:51:41 -07:00
python.py Add an optional Device parameter to pin_memory/is_pinned that does nothing (#60201) 2021-06-28 10:38:52 -07:00
structured.py [structured] Preserve computed elements from meta func to impl (#61746) 2021-09-01 14:34:25 -07:00
translate.py clamp: port to structured kernel (#61361) 2021-07-23 02:02:07 -07:00
types.py clamp: port to structured kernel (#61361) 2021-07-23 02:02:07 -07:00