Home
last modified time | relevance | path

Searched refs:DisableTorchFunctionSubclass (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Ddisable_torch_function.cpp39 } DisableTorchFunctionSubclass; typedef
45 ((DisableTorchFunctionSubclass*)self)->old_state = old_state; in DisableTorchFunctionSubclass__enter()
55 ((DisableTorchFunctionSubclass*)self)->old_state); in DisableTorchFunctionSubclass__exit()
86 sizeof(DisableTorchFunctionSubclass), /* tp_basicsize */
139 ((DisableTorchFunctionSubclass*)self)->old_state = in DisableTorchFunction__enter()
148 ((DisableTorchFunctionSubclass*)self)->old_state); in DisableTorchFunction__exit()
/aosp_15_r20/external/pytorch/torch/distributed/_shard/
H A Dcommon_op_utils.py64 with torch._C.DisableTorchFunctionSubclass():
/aosp_15_r20/external/executorch/exir/
H A Dtracer.py46 from torch._C import _EnableTorchFunction, DisableTorchFunctionSubclass # @manual
289 with DisableTorchFunctionSubclass():
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_subclasses.py515 with torch._C.DisableTorchFunctionSubclass():
526 with torch._C.DisableTorchFunctionSubclass():
527 with torch._C.DisableTorchFunctionSubclass():
539 with torch._C.DisableTorchFunctionSubclass():
555 with torch._C.DisableTorchFunctionSubclass():
1572 with torch._C.DisableTorchFunctionSubclass():
H A Dtest_modes.py321 with BaseTorchFunctionMode(), torch._C.DisableTorchFunctionSubclass():
/aosp_15_r20/external/pytorch/torch/autograd/
H A Dprofiler.py749 with torch._C.DisableTorchFunctionSubclass():
786 with torch._C.DisableTorchFunctionSubclass():
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_load_state_dict.py495 with torch._C.DisableTorchFunctionSubclass():
528 with torch._C.DisableTorchFunctionSubclass():
/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/_ops/
H A Dtensor_ops.py214 with torch._C.DisableTorchFunctionSubclass():
/aosp_15_r20/external/pytorch/test/
H A Dtest_overrides.py1513 with torch._C.DisableTorchFunctionSubclass():
1546 with torch._C.DisableTorchFunctionSubclass():
1566 with torch._C.DisableTorchFunctionSubclass():
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtorch.py79 torch._C.DisableTorchFunctionSubclass,
282 elif self.value is torch._C.DisableTorchFunctionSubclass:
H A Dbuilder.py2135 with torch._C.DisableTorchFunctionSubclass():
/aosp_15_r20/external/pytorch/torch/masked/maskedtensor/
H A Dcore.py295 with torch._C.DisableTorchFunctionSubclass():
/aosp_15_r20/external/pytorch/torch/nested/_internal/
H A Dnested_tensor.py313 with torch._C.DisableTorchFunctionSubclass():
/aosp_15_r20/external/pytorch/torch/
H A D_tensor.py1511 with _C.DisableTorchFunctionSubclass():
/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfake_tensor.py2349 with torch._C.DisableTorchFunctionSubclass():
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in207 def DisableTorchFunctionSubclass(): ...