Home
last modified time | relevance | path

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

/external/pytorch/torch/_dynamo/variables/
Dtorch_function.py174 assert isinstance(var, (TensorWithTFOverrideVariable, UserDefinedObjectVariable))
179 assert isinstance(var, (TensorWithTFOverrideVariable, UserDefinedObjectVariable))
180 if isinstance(var, TensorWithTFOverrideVariable):
265 class TensorWithTFOverrideVariable(TensorVariable): class
Dtensor.py667 from .torch_function import TensorWithTFOverrideVariable
681 return TensorWithTFOverrideVariable.from_tensor_var(
1316 from .torch_function import TensorWithTFOverrideVariable
1322 return TensorWithTFOverrideVariable.from_tensor_var(
Dbuilder.py206 TensorWithTFOverrideVariable,
392 TensorWithTFOverrideVariable,
2039 result = wrap_fx_proxy_cls(target_cls=TensorWithTFOverrideVariable, **kwargs)
2145 in (TensorVariable, TensorWithTFOverrideVariable),
/external/pytorch/torch/_dynamo/
Dcodegen.py34 from .variables.torch_function import TensorWithTFOverrideVariable
158 elif isinstance(value, TensorWithTFOverrideVariable):
Doutput_graph.py107 from .variables.torch_function import TensorWithTFOverrideVariable
1102 TensorWithTFOverrideVariable,
Dutils.py2622 from torch._dynamo.variables.torch_function import TensorWithTFOverrideVariable
2624 if isinstance(vt, TensorWithTFOverrideVariable):
/external/pytorch/test/dynamo/
Dtest_modules.py25 from torch._dynamo.variables.torch_function import TensorWithTFOverrideVariable
1340 original = TensorWithTFOverrideVariable.global_mangled_class_name
1348 TensorWithTFOverrideVariable.global_mangled_class_name = instrumented
1389 TensorWithTFOverrideVariable.global_mangled_class_name = original