code: def default_output_shape(input_shape, kernel, stride, padding):
      if padding == "SAME":
          return input_shape * stride
      elif padding == "VALID":
          return (input_shape - 1) * stride + kernel
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import conv
  from haiku._src import initializers
  from haiku._src import test_utils
  from haiku._src import transform
  import jax
  from jax import random
  import jax.numpy
  import numpy
label_desc: Use image node_id 2 for calling a global function with example usage: default_output_shape(input_shape, kernel, stride, padding) and returns: unknown, unknown