diff options
author | Phil <ijund.phil@gmail.com> | 2018-02-07 19:59:59 +0100 |
---|---|---|
committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2018-02-07 10:59:59 -0800 |
commit | 3d86d8ce14989ca65a59ad4cf37f690694bf6267 (patch) | |
tree | ae2797cd796b292f8303bb58dae23c80489d4749 /ISSUE_TEMPLATE.md | |
parent | 8aa14cd682053e1e643f0a74ec25cf3b87bf2712 (diff) |
Add unsortedsegment(prod/min/max/sqrt_n/mean). (#15858)
* Add unsortedsegment(prod/min/max/sqrt_n/mean).
This commit adds CPU/GPU implementations for prod/min/max
ops and python implementations for mean/sqrt_n. Also, it adapts and unifies the
corresponding tests of all unsorted reductions.
Note: The new gradient of unsorted_segment_max fixes the crash occuring when
negative indices on CPU are used.
* update golden API
* Fix compilation of atomicAdd for cuda_arch < 600. \n This commit moves the std::complex specialization of atomicAdd below the double specialization of atomicAdd for cuda_arch 600.
* Enable bfloat16, change inline to EIGEN_STRONG_INLINE.
* fix includes of cuda_device_functions; fix typo
Diffstat (limited to 'ISSUE_TEMPLATE.md')
0 files changed, 0 insertions, 0 deletions