code: def tall_cases(align=None):
      # pyformat: disable
      mat = np.array(
          [
              [[1, 2, 3], [4, 5, 6], [7, 8, 9], [9, 8, 7], [6, 5, 4]],
              [[3, 2, 1], [1, 2, 3], [4, 5, 6], [7, 8, 9], [9, 8, 7]],
          ]
      )
      tests = dict()
      # tests[d_lower, d_upper] = (compact_diagonals, padded_diagonals)
      tests[0, 0] = (
          np.array([[1, 5, 9], [3, 2, 6]]),
          np.array(
              [
                  [[1, 0, 0], [0, 5, 0], [0, 0, 9], [0, 0, 0]],
                  [[3, 0, 0], [0, 2, 0], [0, 0, 6], [0, 0, 0]],
              ]
          ),
      )
      tests[-4, -3] = (
          np.array([[[9, 5], [6, 0]], [[7, 8], [9, 0]]]),
          np.array(
              [
                  [
                      [0, 0, 0],
                      [0, 0, 0],
                      [0, 0, 0],
                      [9, 0, 0],
                      [6, 5, 0],
                  ],
                  [
                      [0, 0, 0],
                      [0, 0, 0],
                      [0, 0, 0],
                      [7, 0, 0],
                      [9, 8, 0],
                  ],
              ]
          ),
      )
      tests[-2, -1] = (
          np.array([[[4, 8, 7], [7, 8, 4]], [[1, 5, 9], [4, 8, 7]]]),
          np.array(
              [
                  [
                      [0, 0, 0],
                      [4, 0, 0],
                      [7, 8, 0],
                      [0, 8, 7],
                      [0, 0, 4],
                  ],
                  [
                      [0, 0, 0],
                      [1, 0, 0],
                      [4, 5, 0],
                      [0, 8, 9],
                      [0, 0, 7],
                  ],
              ]
          ),
      )
      tests[-2, 1] = (
          np.array(
              [
                  [[2, 6, 0], [1, 5, 9], [4, 8, 7], [7, 8, 4]],
                  [[2, 3, 0], [3, 2, 6], [1, 5, 9], [4, 8, 7]],
              ]
          ),
          np.array(
              [
                  [
                      [1, 2, 0],
                      [4, 5, 6],
                      [7, 8, 9],
                      [0, 8, 7],
                      [0, 0, 4],
                  ],
                  [
                      [3, 2, 0],
                      [1, 2, 3],
                      [4, 5, 6],
                      [0, 8, 9],
                      [0, 0, 7],
                  ],
              ]
          ),
      )
      tests[1, 2] = (
          np.array([[[3, 0], [2, 6]], [[1, 0], [2, 3]]]),
          np.array(
              [
                  [
                      [0, 2, 3],
                      [0, 0, 6],
                      [0, 0, 0],
                      [0, 0, 0],
                      [0, 0, 0],
                  ],
                  [
                      [0, 2, 1],
                      [0, 0, 3],
                      [0, 0, 0],
                      [0, 0, 0],
                      [0, 0, 0],
                  ],
              ]
          ),
      )
      # pyformat: enable
      return (mat, repack_diagonals_in_tests(tests, align))
  
imports: import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gen_array_ops
  from tensorflow.python.platform import googletest
label_desc: Use image node_id 5 for calling a global function with example usage: tall_cases(align) and returns: mat,  repack_diagonals_in_tests