Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtorch_function.py19 from .tensor import TensorSubclassVariable, TensorVariable
300 return TensorSubclassVariable(
H A Dtensor.py664 if isinstance(cls, TensorSubclassVariable) and cls.source:
1303 class TensorSubclassVariable(VariableTracker): class
H A Dbuilder.py199 TensorSubclassVariable,
870 return TensorSubclassVariable(value, source=self.source)