@test_util.run_deprecated_v1
  def testGradientV2SegmentsInvalid1(self):
    tf_x, _ = self._input(
        [3, 4], dtype=dtypes_lib.float32
    )  # expecting 3 segments