Searched refs:tensor_variable (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | torch.py | 897 tensor_variable = wrap_fx_proxy(
907 isinstance(tensor_variable, TensorVariable)
923 if isinstance(tensor_variable, TupleVariable):
930 tx.symbolic_locals[name] = tensor_variable.items[idx]
932 kwargs["out"].items, tensor_variable.items
944 elif isinstance(tensor_variable, TensorVariable):
947 fake_tensor = tensor_variable.proxy.node.meta["example_value"]
965 tx.symbolic_locals[name] = tensor_variable
967 isinstance(tensor_variable, ConstantVariable)
968 and tensor_variable.value is None
[all …]
|
| H A D | tensor.py | 1275 def from_tensor_variable(cls, tensor_variable, raw_value, need_unwrap=True): argument
1278 **dict(tensor_variable.__dict__),
1299 def from_tensor_variable(cls, tensor_variable): argument
1300 return FakeItemVariable(**dict(tensor_variable.__dict__))
|
| H A D | builder.py | 1243 for i, tensor_variable in enumerate(list_variable.items):
1246 self.tx.output.input_source_to_var[source_i] = tensor_variable
1247 tensor_variable.proxy.node.meta["tensor_dict"] = _extract_tensor_dict(
1593 tensor_variable = wrap_fx_proxy(
1635 self.tx.output.input_source_to_var[source] = tensor_variable
1640 fake_tensor_value = tensor_variable.proxy.node.meta["example_value"]
1647 return tensor_variable
|
| H A D | user_defined.py | 538 tensor_variable = wrap_fx_proxy(
547 return tensor_variable
|
| H A D | builtin.py | 1972 tensor_variable = args[0]
1973 return tensor_variable.call_id(tx)
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | symbolic_convert.py | 2513 def find_symbolic_locals_name(self, tensor_variable): argument
2515 if value is tensor_variable:
|