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
File | Mode | Size |
---|---|---|
benchmarks | ||
api_test.py | -rw-r--r-- | 92.9 KB |
array_interoperability_test.py | -rw-r--r-- | 4.5 KB |
batching_test.py | -rw-r--r-- | 39.2 KB |
core_test.py | -rw-r--r-- | 7.9 KB |
debug_nans_test.py | -rw-r--r-- | 1.5 KB |
dtypes_test.py | -rw-r--r-- | 7.0 KB |
fft_test.py | -rw-r--r-- | 14.4 KB |
generated_fun_test.py | -rw-r--r-- | 7.7 KB |
infeed_test.py | -rw-r--r-- | 2.8 KB |
jax_to_hlo_test.py | -rw-r--r-- | 2.3 KB |
jet_test.py | -rw-r--r-- | 7.4 KB |
lax_control_flow_test.py | -rw-r--r-- | 59.3 KB |
lax_numpy_einsum_test.py | -rw-r--r-- | 8.4 KB |
lax_numpy_indexing_test.py | -rw-r--r-- | 39.1 KB |
lax_numpy_test.py | -rw-r--r-- | 129.2 KB |
lax_numpy_vectorize_test.py | -rw-r--r-- | 7.5 KB |
lax_scipy_sparse_test.py | -rw-r--r-- | 4.9 KB |
lax_scipy_test.py | -rw-r--r-- | 7.3 KB |
lax_test.py | -rw-r--r-- | 148.4 KB |
linalg_test.py | -rw-r--r-- | 48.2 KB |
loops_test.py | -rw-r--r-- | 12.7 KB |
masking_test.py | -rw-r--r-- | 15.8 KB |
metadata_test.py | -rw-r--r-- | 3.8 KB |
multi_device_test.py | -rw-r--r-- | 7.1 KB |
multibackend_test.py | -rw-r--r-- | 5.3 KB |
nn_test.py | -rw-r--r-- | 6.2 KB |
optimizers_test.py | -rw-r--r-- | 9.7 KB |
optix_test.py | -rw-r--r-- | 5.1 KB |
parallel_test.py | -rw-r--r-- | 9.5 KB |
pmap_test.py | -rw-r--r-- | 38.0 KB |
profiler_test.py | -rw-r--r-- | 1.7 KB |
random_test.py | -rw-r--r-- | 20.4 KB |
scipy_ndimage_test.py | -rw-r--r-- | 4.8 KB |
scipy_stats_test.py | -rw-r--r-- | 16.0 KB |
stax_test.py | -rw-r--r-- | 10.7 KB |
tree_util_tests.py | -rw-r--r-- | 5.1 KB |
util_test.py | -rw-r--r-- | 2.5 KB |
vectorize_test.py | -rw-r--r-- | 5.1 KB |
xla_bridge_test.py | -rw-r--r-- | 1.6 KB |
Computing file changes ...