Home
last modified time | relevance | path

Searched refs:unwrap_proxy (Results 1 – 19 of 19) sorted by relevance

/external/pytorch/torch/_prims/
Drng_prims.py208 proxy_args = pytree.tree_map(mode.tracer.unwrap_proxy, (op, *args))
209 proxy_kwargs = pytree.tree_map(mode.tracer.unwrap_proxy, kwargs)
272 proxy_args = pytree.tree_map(mode.tracer.unwrap_proxy, (rng_state, op, *args))
273 proxy_kwargs = pytree.tree_map(mode.tracer.unwrap_proxy, kwargs)
/external/pytorch/torch/_higher_order_ops/
Deffects.py172 proxy_token = mode.tracer.unwrap_proxy(token)
173 proxy_args = pytree.tree_map(mode.tracer.unwrap_proxy, args)
174 proxy_kwargs = pytree.tree_map(mode.tracer.unwrap_proxy, kwargs)
Dtorchbind.py80 proxy_args = pytree.tree_map(mode.tracer.unwrap_proxy, args)
81 proxy_kwargs = pytree.tree_map(mode.tracer.unwrap_proxy, kwargs)
Dhints_wrap.py133 proxy_args = pytree.tree_map(proxy_mode.tracer.unwrap_proxy, new_args)
134 proxy_kwargs = pytree.tree_map(proxy_mode.tracer.unwrap_proxy, new_kwargs)
Drun_const_graph.py25 p_args = pytree.tree_map(mode.tracer.unwrap_proxy, args)
Dstrict_mode.py70 proxy_args = pytree.tree_map(mode.tracer.unwrap_proxy, args)
Dout_dtype.py94 proxy_args = pytree.tree_map(proxy_mode.tracer.unwrap_proxy, node_args)
Dwhile_loop.py210 proxy_args = pytree.tree_map(proxy_mode.tracer.unwrap_proxy, args)
Dauto_functionalize.py602 proxy_kwargs = pytree.tree_map(mode.tracer.unwrap_proxy, kwargs)
697 proxy_kwargs = pytree.tree_map(mode.tracer.unwrap_proxy, kwargs)
Dmap.py207 proxy_args = pytree.tree_map(proxy_mode.tracer.unwrap_proxy, node_args)
Dassociative_scan.py325 proxy_args = pytree.tree_map(proxy_mode.tracer.unwrap_proxy, args)
Dcond.py310 proxy_args = pytree.tree_map(proxy_mode.tracer.unwrap_proxy, args)
Dflex_attention.py349 proxy_args = pytree.tree_map(proxy_mode.tracer.unwrap_proxy, node_args)
901 proxy_args = pytree.tree_map(proxy_mode.tracer.unwrap_proxy, node_args)
Dtriton_kernel_wrap.py597 proxy_args = pytree.tree_map(proxy_mode.tracer.unwrap_proxy, node_args)
/external/executorch/exir/
Dcontrol_flow.py64 unwrap_proxy,
165 proxies = tuple([unwrap_proxy(v) for v in flattened_inputs])
Dtracer.py179 def unwrap_proxy(t: LeafValue) -> Union[LeafValue, torch.fx.Proxy]: function
328 proxy_args = ex_pytree.tree_map(unwrap_proxy, args)
330 proxy_kwargs = ex_pytree.tree_map(unwrap_proxy, kwargs)
536 return unwrap_proxy(out)
/external/pytorch/torch/_dynamo/
D_trace_wrapped_higher_order_op.py78 return mode.tracer.unwrap_proxy(x)
/external/pytorch/torch/_export/
Dwrappers.py28 p_args, p_kwargs = pytree.tree_map(mode.tracer.unwrap_proxy, (args, kwargs))
/external/pytorch/torch/fx/experimental/
Dproxy_tensor.py1063 def unwrap_proxy(self, e: Tensor) -> Union[Proxy, Tensor]: member in PythonKeyTracer
1067 def unwrap_proxy(self, e: PySymType) -> Union[Proxy, PySymType]: member in PythonKeyTracer
1071 def unwrap_proxy( member in PythonKeyTracer
1076 def unwrap_proxy(self, e: T) -> object: member in PythonKeyTracer