Revision cf4dd84b1452a089a85497fe66915dfda27508f3 authored by Peter Hawkins on 06 April 2020, 19:14:22 UTC, committed by GitHub on 06 April 2020, 19:14:22 UTC
* cumsum is linear, so its gradient can be linear also.

* Rename _impl functions to _prefix_scan.
1 parent fc23e07
History
File Mode Size
BUILD -rw-r--r-- 5.2 KB
cublas.cc -rw-r--r-- 12.7 KB
cuda_prng.py -rw-r--r-- 1.7 KB
cuda_prng_kernels.cc -rw-r--r-- 1.2 KB
cuda_prng_kernels.cu.cc -rw-r--r-- 4.2 KB
cuda_prng_kernels.h -rw-r--r-- 1.0 KB
cusolver.cc -rw-r--r-- 46.4 KB
cusolver.py -rw-r--r-- 13.7 KB
gpu_kernel_helpers.cc -rw-r--r-- 1.5 KB
gpu_kernel_helpers.h -rw-r--r-- 1.3 KB
kernel_helpers.h -rw-r--r-- 1.6 KB
kernel_pybind11_helpers.h -rw-r--r-- 1.6 KB
lapack.pyx -rw-r--r-- 58.8 KB
pytree.cc -rw-r--r-- 26.4 KB
version.py -rw-r--r-- 599 bytes

back to top