jax
6cd995e3 - allow tokens in op-by-op by calling into _xla_callable_args

Commit

6 years ago

allow tokens in op-by-op by calling into _xla_callable_args

References

#1667 - simplify [p]xla.py by in-lining single-caller function

Author

mattjj

mattjj

Parents

Loading