projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 39ba738) | patch
Add complex support for tf.segment_mean (#19225)
authorYong Tang <yong.tang.github@outlook.com>
Mon, 14 May 2018 18:10:27 +0000 (11:10 -0700)
committerRasmus Munk Larsen <rmlarsen@google.com>
Mon, 14 May 2018 18:10:27 +0000 (11:10 -0700)
commit0bb7a191a33222c44ff50a3c74b550ee72f8b0e4
treeabcac4860e8a82dfb59e89c7d416b77dddd81f77tree | snapshot
parent39ba73897cd3a5e14d3e78624f0b5942479f533acommit | diff
Add complex support for tf.segment_mean (#19225)

* Add complex support for tf.segment_mean

While using tf.segment_mean I noticed that it does not
have the complex support like tf.segment_sum. I think it
makes sense to support complex for it. This fix adds the
complex support for tf.segment_mean.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add test cases for complex support with tf.segment_mean

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
tensorflow/core/kernels/segment_reduction_ops.cc diff | blob | history
tensorflow/core/ops/math_ops.cc diff | blob | history
tensorflow/python/kernel_tests/segment_reduction_ops_test.py diff | blob | history
Domain: Machine Learning / ML Framework;