Short name describing what triggered the graph break
torch.associative_scan: mismatched input/output tree structure
Values or code snippet captured at the break point
xs: {xs_treespec.as_python_constant()}, output: {_combine_treespec.as_python_constant()}
Explanation of why the graph break was triggered
The tree structure of the xs and the outs of the combine_fn are are expected to be identical, but got xs: {xs_treespec.as_python_constant()} vs output: {_combine_treespec.as_python_constant()}.
Hints on how to resolve the graph break