Shangdi Yu aa0352ca38 [custom ops] add default value support for device types (#129792) ... Fixes #129371 I think the first case in Issue #129371 is already supported in the current code? Since it takes care of string default values. This PR adds support for device type default values. Pull Request resolved: https://github.com/pytorch/pytorch/pull/129792 Approved by: https://github.com/zou3519		2024-07-02 23:31:29 +00:00
..
__init__.py	Rename impl_abstract to register_fake, part 2/2 (#123938)	2024-06-14 14:37:24 +00:00
autograd.py	[custom_op] don't invoke autograd.Function when unnecessary (#127976)	2024-06-13 23:38:23 +00:00
custom_ops.py	[custom ops] Add unknown arg (#129614)	2024-07-02 16:10:14 +00:00
fake_class_registry.py	[torchbind] support tensor ops inside of __obj_flatten__ (#129605)	2024-06-27 03:07:31 +00:00
fake_impl.py	Rename impl_abstract to register_fake, part 2/2 (#123938)	2024-06-14 14:37:24 +00:00
infer_schema.py	[custom ops] add default value support for device types (#129792)	2024-07-02 23:31:29 +00:00
simple_registry.py	Rename impl_abstract to register_fake, part 2/2 (#123938)	2024-06-14 14:37:24 +00:00
utils.py	[BE] enable UFMT for torch/nn/*.py (#128593)	2024-06-23 16:05:13 +00:00