Searched refs:ex_pytree (Results 1 – 3 of 3) sorted by relevance
14 import executorch.extension.pytree as ex_pytree namespace372 args, pytree = ex_pytree.tree_flatten((raw_args, {}))423 treespec = ex_pytree.from_str(self.container_metatype.encoded_out_str)425 return ex_pytree.tree_unflatten(ret, treespec)
27 import executorch.extension.pytree as ex_pytree namespace328 proxy_args = ex_pytree.tree_map(unwrap_proxy, args)330 proxy_kwargs = ex_pytree.tree_map(unwrap_proxy, kwargs)
40 import executorch.extension.pytree as ex_pytree namespace1226 flattened_output, spec = ex_pytree.tree_flatten(vals)1486 _, tree = ex_pytree.tree_flatten(tree)