Searched defs:unflatten_args (Results 1 – 2 of 2) sorted by relevance
4128 def unflatten_args(new_tensor_args, new_non_tensor_args): function5222 unflatten_args, argument5718 unflatten_args, argument7866 unflatten_args, argument
1231 auto unflatten_args = tree_flatten(A, args, flat_args); in run_torch_function() local