Home
last modified time | relevance | path

Searched refs:capture_func_transforms (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_higher_order_ops.py2871 @config.patch(capture_func_transforms=True)
3212 with config.patch(capture_func_transforms=False):
3474 with config.patch(capture_func_transforms=False):
3708 with config.patch(capture_func_transforms=False):
3848 with config.patch(capture_func_transforms=False):
4550 with config.patch(capture_func_transforms=False):
4940 with config.patch(capture_func_transforms=False):
5380 with config.patch(capture_func_transforms=False):
5401 @config.patch(capture_func_transforms=True)
5459 with config.patch(capture_func_transforms=False):
[all …]
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dconfig.py428 capture_func_transforms = True variable
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dhigher_order_ops.py1181 if not torch._dynamo.config.capture_func_transforms: