Commit 5cc56829 authored by Yong Tang's avatar Yong Tang
Browse files

Add complex numbers to the supported data types for UnsortedSegmentProd



In the kernel implementation both UnsortedSegmentProd and UnsortedSegmentSum
supports complex numbers. However, unlike UnsortedSegmentSum, the op
of UnsortedSegmentProd does not register complex number types in math_ops.cc.

This fix adds the supported complex number types to math_ops.cc,
and enables test cases for it.

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