Home
last modified time | relevance | path

Searched refs:torch_function_enabled (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Ddisable_torch_function.cpp13 bool torch_function_enabled() { in torch_function_enabled() function
60 if (torch::torch_function_enabled()) { in THPModule_isEnabledTorchFunction()
322 torch::torch_function_enabled() && has_torch_function_attr(obj)); in check_has_torch_function()
H A Ddisable_torch_function.h11 bool torch_function_enabled();
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtorch_function.py234 return tx.output.torch_function_enabled and any(
327 if tx.output.torch_function_enabled and hasattr(torch.Tensor, name):
366 if tx.output.torch_function_enabled:
H A Dctx_manager.py621 initial_values=[tx.output.torch_function_enabled],
H A Dtorch.py480 return ConstantVariable.create(tx.output.torch_function_enabled)
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Doutput_graph.py367 self.torch_function_enabled = torch._C._is_torch_function_enabled()
616 self.torch_function_enabled,
1700 self.torch_function_enabled = enabled
/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dguards.cpp516 _torch_function = torch::torch_function_enabled(); in init()
530 _torch_function == torch::torch_function_enabled() && in check()
548 if (_torch_function != torch::torch_function_enabled()) in reason()