message: Add additional test to cover squared difference for complex
  where imag parts are not 0.
diff: @@ -213,6 +213,16 @@ class SquaredDifferenceTest(test_util.TensorFlowTestCase):
  z_tf = self.evaluate(math_ops.squared_difference(x, y))
  self.assertAllClose(z, z_tf)
  + @test_util.run_in_graph_and_eager_modes()
  + def testComplexSquaredDifference(self):
  + for dtype in [np.complex64, np.complex128]:
  + x = np.array([[1+3j, 2+2j, 3+1j], [4-1j, 5-2j, 6-3j]], dtype=dtype)
  + y = np.array([-3+1j, -2+2j, -1+3j], dtype=dtype)
  + z = (x - y) * (x - y)
  + with test_util.device(use_gpu=True):
  + z_tf = self.evaluate(math_ops.squared_difference(x, y))
  + self.assertAllClose(z, z_tf)
  +
  @test_util.with_c_api
  class ApproximateEqualTest(test_util.TensorFlowTestCase):
  