move control flow batching tests - SemanticDiff

Commit

7 years ago

move control flow batching tests

References

#755 - add vmap support to `lax.scan` (so vmap-of-scan works)

Author

mattjj

mattjj

Committer

mattjj

mattjj

Parents

Loading