Searched refs:wrap_with_proxy (Results 1 – 2 of 2) sorted by relevance
| /external/executorch/exir/ |
| D | tracer.py | 357 def wrap_with_proxy(e: LeafValue, proxy: torch.fx.Proxy) -> LeafValue: function
379 retval = single_return(real_out, proxy_out, wrap_with_proxy)
381 retval = tree_return(real_out, proxy_out, wrap_with_proxy, type(real_out))
|
| /external/pytorch/torch/fx/experimental/ |
| D | proxy_tensor.py | 614 def wrap_with_proxy( function
648 wrap_with_proxy(ee, proxy[idx], get_constant(constant, idx)) # type: ignore[index]
664 wrap_with_proxy(val, proxy[key], None) # type: ignore[index]
674 wrap_with_proxy(inner_res, proxy_res, constant)
|