jax
cf4dd84b - cumsum is linear, so its gradient can be linear also. (#2618)

Commit

5 years ago

cumsum is linear, so its gradient can be linear also. (#2618) * cumsum is linear, so its gradient can be linear also. * Rename _impl functions to _prefix_scan.

References

#2618 - cumsum is linear, so its gradient can be linear also.

Author

hawkinsp

Parents

fc23e071

jax cf4dd84b - cumsum is linear, so its gradient can be linear also. (#2618)

jax
cf4dd84b - cumsum is linear, so its gradient can be linear also. (#2618)