Home
last modified time | relevance | path

Searched refs:TorchInGraphFunctionVariable (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtrace_rules.py53 TorchInGraphFunctionVariable,
137 "torch.onnx.is_in_onnx_export": TorchInGraphFunctionVariable,
138 "torch.onnx.operators.shape_as_tensor": TorchInGraphFunctionVariable,
139 "torch.overrides.is_tensor_like": TorchInGraphFunctionVariable,
140 "torch.jit.is_scripting": TorchInGraphFunctionVariable,
141 "torch.jit.is_tracing": TorchInGraphFunctionVariable,
142 "torch.jit.annotate": TorchInGraphFunctionVariable,
143 "torch.distributed.is_available": TorchInGraphFunctionVariable,
144 "torch.distributed.is_initialized": TorchInGraphFunctionVariable,
145 "torch.distributed.get_rank": TorchInGraphFunctionVariable,
[all …]
H A Ddecorators.py89 if trace_rules.lookup_callable(fn) != variables.TorchInGraphFunctionVariable:
103 != variables.TorchInGraphFunctionVariable
104 and trace_rules.lookup(fn) != variables.TorchInGraphFunctionVariable
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtensor.py329 return variables.TorchInGraphFunctionVariable(
342 return variables.TorchInGraphFunctionVariable(_tensor_version).call_function(
867 result = variables.TorchInGraphFunctionVariable(torch.mul).call_function(
877 result = variables.TorchInGraphFunctionVariable(torch.div).call_function(
880 result = variables.TorchInGraphFunctionVariable(torch.mul).call_function(
894 result = variables.TorchInGraphFunctionVariable(torch.eq).call_function(
897 result = variables.TorchInGraphFunctionVariable(torch.any).call_function(
H A Dsdpa.py25 from .torch import TorchInGraphFunctionVariable
49 return TorchInGraphFunctionVariable(SDPAParams).call_function(
H A Dtorch.py331 class TorchInGraphFunctionVariable(BaseTorchVariable): class
569 result = TorchInGraphFunctionVariable(torch.div).call_function(
572 result = TorchInGraphFunctionVariable(torch.mul).call_function(
575 return TorchInGraphFunctionVariable(torch.add).call_function(
749 return TorchInGraphFunctionVariable(
776 return TorchInGraphFunctionVariable(torch._refs.tensor).call_function(
H A Dbuiltin.py1088 a = variables.TorchInGraphFunctionVariable(torch.tensor).call_function(
1113 fn = variables.TorchInGraphFunctionVariable(torch.clamp)
1124 fn = variables.TorchInGraphFunctionVariable(fn)
1642 TorchInGraphFunctionVariable,
1730 elif isinstance(obj, TorchInGraphFunctionVariable):
1736 return TorchInGraphFunctionVariable(member, **options)
H A D__init__.py104 from .torch import TorchCtxManagerClassVariable, TorchInGraphFunctionVariable
H A Duser_defined.py892 == variables.TorchInGraphFunctionVariable
922 return variables.TorchInGraphFunctionVariable(
H A Dctx_manager.py909 return variables.TorchInGraphFunctionVariable(
H A Ddicts.py55 variables.TorchInGraphFunctionVariable,
H A Dbuilder.py203 from .torch import TorchCtxManagerClassVariable, TorchInGraphFunctionVariable
2194 return TorchInGraphFunctionVariable(proxy.node.target)
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_trace_rules.py24 from torch._dynamo.variables import TorchInGraphFunctionVariable, UserFunctionVariable
151 ] = TorchInGraphFunctionVariable
/aosp_15_r20/external/pytorch/docs/source/
H A Dtorch.compiler_dynamo_deepdive.rst300 TRACE LOAD_GLOBAL y [TorchInGraphFunctionVariable(<built-in method any>), TensorVariable()]