code: def IdentityBlock(
      kernel_size, filters, norm, non_linearity, mode="train"
  ):
      """ResNet identical size block."""
      ks = kernel_size
      filters1, filters2, filters3 = filters
      main = [
          tl.Conv(filters1, (1, 1)),
          norm(mode=mode),
          non_linearity(),
          tl.Conv(filters2, (ks, ks), padding="SAME"),
          norm(mode=mode),
          non_linearity(),
          tl.Conv(filters3, (1, 1)),
          norm(mode=mode),
      ]
      return [
          tl.Residual(main),
          non_linearity(),
      ]
  
imports: from trax import layers
label_desc: Use image node_id 2 for calling a global function with example usage: IdentityBlock(kernel_size, filters, norm, non_linearity, mode) and returns: list