Searched refs:unflatten_args (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/_inductor/ |
| D | ir.py | 4128 def unflatten_args(new_tensor_args, new_non_tensor_args): function
4164 new_args, new_kwargs = unflatten_args(example_args, non_tensor_args)
4190 unflatten_args,
5222 unflatten_args, argument
5259 self.unflatten_args = unflatten_args
5305 args, kwargs = self.unflatten_args(self.inputs, self.constant_args)
5430 args, kwargs = self.unflatten_args(tensor_args, self.constant_args)
5483 args, kwargs = self.unflatten_args(self.inputs, self.constant_args)
5637 unflatten_args,
5648 unflatten_args,
[all …]
|
| /external/pytorch/functorch/csrc/dim/ |
| D | dim.cpp | 1231 auto unflatten_args = tree_flatten(A, args, flat_args); in run_torch_function() local
1265 mpy::vector_args uargs = unflatten_args(A, flat_it); in run_torch_function()
1295 mpy::vector_args uargs = unflatten_args(A, flat_it); in run_torch_function()
|