Commit 21a9efc4 authored by Yong Tang's avatar Yong Tang
Browse files

Add complex dtypes support for `tf.squared_difference`



This fix tries to address the issue raised in 14932 where
complex dtypes are not supported for `tf.squared_difference`,
which is different from the doc string in `math_ops.cc`
(see `BINARY_FEWER`).

This fix adds the complex64 and complex128 support in kernel,
and adds additional test cases.

This fix fixes 14932.

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