mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove trailing whitespace

Browse Source
This commit is contained in:
Erich Elsen 2020-06-28 21:33:42 +01:00
parent d7e3495d27
commit b8d0de6365
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/lax/lax.py
View File

@ -4915,7 +4915,7 @@ def _cumulative_reduction_primitive(name, prefix_scan_fn, jvp_rule, reduce_windo
reducer_p = standard_primitive(
_cumred_shape_rule, partial(_reduce_number_dtype_rule, name),
name, xla.lower_fun(prefix_scan_fn, multiple_results=False))
ad.primitive_jvps[reducer_p] = jvp_rule
ad.primitive_jvps[reducer_p] = jvp_rule
xla.backend_specific_translations['tpu'][reducer_p] = xla.lower_fun(
partial(_cumred_tpu_translation_rule, reduce_window_fn),
multiple_results=False)