Home
last modified time | relevance | path

Searched refs:tuple_cls (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dlists.py43 return functools.partial(NamedTupleVariable, tuple_cls=type(obj))
715 def __init__(self, items, tuple_cls, **kwargs) -> None: argument
717 self.tuple_cls = tuple_cls
720 return repr(self.tuple_cls(*(Lit(x.debug_repr()) for x in self.items)))
723 return self.tuple_cls
733 create_fn = getattr(self.tuple_cls, "_make", self.tuple_cls)
747 method = inspect.getattr_static(self.tuple_cls, name, None)
752 variables.UserDefinedClassVariable(self.tuple_cls),
761 fields = namedtuple_fields(self.tuple_cls)
770 return variables.ConstantVariable.create(hasattr(self.tuple_cls, name))
/aosp_15_r20/external/pytorch/torch/_custom_op/
H A Dautograd.py166 tuple_cls = namedtuple(name, attribs) # type: ignore[misc]
167 return tuple_cls
172 tuple_cls = namedtuple_args_cls(schema)
173 return tuple_cls(*args)