Commit 0bb7a191 authored by Yong Tang's avatar Yong Tang Committed by Rasmus Munk Larsen
Browse files

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: default avatarYong Tang <yong.tang.github@outlook.com>

* Add test cases for complex support with tf.segment_mean

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent 39ba7389
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment