pytorch/torch/_library
2024-05-07 03:02:00 +00:00
..
__init__.py Add register_autograd to register backward formulas for custom ops (#123110) 2024-04-05 22:03:47 +00:00
abstract_impl.py [dynamo] support torchbind object input (#124978) 2024-05-07 03:02:00 +00:00
autograd.py [custom_op] setup_context fills in default values (#124852) 2024-04-25 04:22:01 +00:00
custom_ops.py Fix torch.library.register_fake's module reporting (#125037) 2024-04-26 20:53:33 +00:00
fake_class_registry.py [dynamo] support torchbind object input (#124978) 2024-05-07 03:02:00 +00:00
infer_schema.py [custom_op] fix schema inference for kwarg-only args (#124637) 2024-04-25 01:51:02 +00:00
simple_registry.py Rename impl_abstract to register_fake, part 1/2 (#123937) 2024-04-17 12:46:01 +00:00
utils.py [custom_op] setup_context fills in default values (#124852) 2024-04-25 04:22:01 +00:00