jax
7c5346f2 - [fuser] Fix support for single-output stateful fusions.

Commit
8 days ago
[fuser] Fix support for single-output stateful fusions. In some cases, a fusion might return `(value, None)` or `(None, value)` which is still a single output, but the check wrongly disallows it since it counts those as 2 outputs. We filter out `None`s in the output to support this. PiperOrigin-RevId: 964763891
Author
Parents
Loading