message: Updated the error type raised when using `tf.linalg.diag_part`.
diff: @@ -2697,6 +2697,9 @@ def matrix_diag_part(
  Returns:
  A Tensor containing diagonals of `input`. Has the same type as `input`.
  +
  + Raises:
  + InvalidArgumentError: When `k` is out of bound or when `k[0]>k[1:]`.
  """
  # Special case to sidestep the tf.constant conversion error:
  # TypeError: Expected bool, got 0 of type 'int' instead.
  