| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | iter.py | 50 seqs = [arg.unpack_var_sequence(tx) for arg in args]
67 seq = args[0].unpack_var_sequence(tx)
107 iterable = args[0].unpack_var_sequence(tx)
132 seq = args[0].unpack_var_sequence(tx)
354 def unpack_var_sequence(self, tx) -> List["VariableTracker"]: member in ZipVariable
361 iterables.append(it.unpack_var_sequence(tx))
|
| H A D | base.py | 289 def unpack_var_sequence(self, tx) -> List["VariableTracker"]: member in VariableTracker
299 return self.unpack_var_sequence(tx)
303 self.unpack_var_sequence(tx)
335 return variables.ConstantVariable.create(len(self.unpack_var_sequence(tx)))
|
| H A D | higher_order_ops.py | 156 ).unpack_var_sequence(tx)
159 flat_args.unpack_var_sequence(tx),
168 ).unpack_var_sequence(tx)
434 ).unpack_var_sequence(tx)
611 operands = args[3].unpack_var_sequence(tx)
695 true_r.unpack_var_sequence(tx), false_r.unpack_var_sequence(tx)
820 operands = args[2].unpack_var_sequence(tx)
831 additional_inputs = args[3].unpack_var_sequence(tx)
1305 unpacked_sequence = args[1].unpack_var_sequence(tx)
|
| H A D | builtin.py | 369 return TupleVariable([*a.items, *b.unpack_var_sequence(tx)])
372 return SizeVariable([*a.items, *b.unpack_var_sequence(tx)])
393 [*a.unpack_var_sequence(tx), *b.items],
402 [*a.items, *b.unpack_var_sequence(tx)], mutable_local=MutableLocal()
417 seq = b.unpack_var_sequence(tx)
1261 list(obj.unpack_var_sequence(tx)),
1453 items = arg.unpack_var_sequence(tx)
1463 arg.unpack_var_sequence(tx) if arg.has_unpack_var_sequence(tx) else arg
1568 seq.unpack_var_sequence(tx) if seq.has_unpack_var_sequence(tx) else seq
1575 seq_unpacked = seq.unpack_var_sequence(tx)
[all …]
|
| H A D | lists.py | 109 def unpack_var_sequence(self, tx): member in BaseListVariable
135 return iter_contains(self.unpack_var_sequence(tx), args[0], tx)
296 def unpack_var_sequence(self, tx=None): member in RangeVariable
644 def unpack_var_sequence(self, tx): member in SizeVariable
868 def unpack_var_sequence(self, tx): member in ListIteratorVariable
872 return self.unpack_var_sequence(tx)
|
| H A D | constant.py | 100 return self.unpack_var_sequence(tx=None)
116 def unpack_var_sequence(self, tx): member in ConstantVariable
|
| H A D | nn_module.py | 158 def unpack_var_sequence(self, tx): member in NNModuleVariable
826 def unpack_var_sequence(self, tx): member in UnspecializedNNModuleVariable
846 ).unpack_var_sequence(tx)
848 return super().unpack_var_sequence(tx)
|
| H A D | dicts.py | 356 def unpack_var_sequence(self, tx): member in ConstDictVariable
521 arg = SetVariable(args[0].unpack_var_sequence(tx))
616 def unpack_var_sequence(self, tx): member in DictView
|
| H A D | optimizer.py | 260 zip(group["params"], params_vt.unpack_var_sequence(tx))
|
| H A D | torch.py | 489 args[0].unpack_var_sequence(tx)
1007 list(value.unpack_var_sequence(tx)),
|
| H A D | user_defined.py | 814 def unpack_var_sequence(self, tx): member in UserDefinedObjectVariable
829 return super().unpack_var_sequence(tx)
|
| H A D | tensor.py | 468 def unpack_var_sequence(self, tx: "InstructionTranslator", idxes=None): member in TensorVariable
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | symbolic_convert.py | 530 if truth_fn(len(value.unpack_var_sequence(self))):
1959 val = seq.unpack_var_sequence(self, idxes=range(inst.argval)) # type: ignore[arg-type]
|