(t: LeafValue)
| 178 | |
| 179 | |
| 180 | def unwrap_proxy(t: LeafValue) -> Union[LeafValue, torch.fx.Proxy]: |
| 181 | if not isinstance(t, torch.Tensor): |
| 182 | return t |
| 183 | t = unwrap_functional(t) |
| 184 | return t.proxy if isinstance(t, PythonTensor) else t |
| 185 | |
| 186 | |
| 187 | def single_return( |
no test coverage detected