Searched refs:user_cls (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/torch/_dynamo/variables/ |
| D | dicts.py | 131 self, items: Dict[VariableTracker, VariableTracker], user_cls=dict, **kwargs argument
148 self.user_cls = user_cls
172 return self.user_cls
194 if self.user_cls is collections.OrderedDict:
208 if self.user_cls is collections.OrderedDict:
362 if self.user_cls is dict:
363 if name in self.user_cls.__dict__:
370 def __init__(self, items, user_cls, default_factory=None, **kwargs) -> None: argument
371 super().__init__(items, user_cls, **kwargs)
372 assert user_cls is collections.defaultdict
[all …]
|
| D | lists.py | 918 user_cls: type,
923 not istype(user_cls, type)
924 or user_cls.__bases__ != (python_cls,)
925 or user_cls.__mro__ != (user_cls, python_cls, object)
930 for name in set(user_cls.__dict__.keys()) - cls._allowed_names
934 def is_matching_cls(cls, user_cls: type):
935 return cls._is_non_conflicting_subclass(user_cls, list)
938 self, items, *, user_cls: type, user_cls_source: Source, **kwargs
941 self.user_cls = user_cls
943 assert istype(user_cls, type)
[all …]
|
| D | script_object.py | 31 def is_matching_cls(cls, user_cls: type):
32 return issubclass(user_cls, torch.ScriptObject)
|
| D | builtin.py | 1330 def call_custom_dict(tx: "InstructionTranslator", user_cls, *args, **kwargs): argument
1339 return ConstDictVariable(arg, user_cls, mutable_local=MutableLocal())
1341 return arg.clone(user_cls=user_cls, mutable_local=MutableLocal())
1355 return ConstDictVariable(items, user_cls, mutable_local=MutableLocal())
1377 return BuiltinVariable(user_cls).call_custom_dict(tx, user_cls, out)
1381 items, user_cls=user_cls, mutable_local=MutableLocal()
1387 tx: "InstructionTranslator", user_cls, *args, **kwargs argument
1389 assert user_cls in {dict, OrderedDict, defaultdict}
1392 assert user_cls is OrderedDict
1402 ConstDictVariable if user_cls is not defaultdict else DefaultDictVariable
[all …]
|
| D | builder.py | 1092 user_cls=type(value),
2012 value = obj.user_cls
|
| D | user_defined.py | 509 user_cls=self.value,
|
| D | misc.py | 418 user_cls=dict,
|
| /external/pytorch/torch/_dynamo/ |
| D | side_effects.py | 254 user_cls: Any,
258 if user_cls is torch.autograd.function.FunctionCtx:
261 elif issubclass(user_cls, torch.nn.Module):
262 obj = nn_module_new(user_cls)
264 obj = object_new(user_cls)
279 user_cls = cls_variable.value
285 if issubclass(user_cls, torch.nn.Module):
287 elif issubclass(user_cls, MutableMapping):
289 elif is_frozen_dataclass(user_cls):
298 return self.track_object_new(cls_source, user_cls, variable_cls, {})
|