Searched refs:need_unwrap (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | tensor.py | 1268 self, proxy: torch.fx.Proxy, *, raw_value=None, need_unwrap=True, **kwargs argument
1272 self.need_unwrap = need_unwrap
1275 def from_tensor_variable(cls, tensor_variable, raw_value, need_unwrap=True): argument
1280 need_unwrap=need_unwrap,
1294 need_unwrap = kwargs.pop("need_unwrap", False)
1296 self.need_unwrap = need_unwrap
|
| H A D | builtin.py | 921 need_unwrap = any(
922 x.need_unwrap
932 need_unwrap=need_unwrap,
1150 need_unwrap = any(
1151 x.need_unwrap
1156 result, raw_res, need_unwrap
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | codegen.py | 208 elif isinstance(value, UnspecializedPythonVariable) and value.need_unwrap:
|