code: def _conv3x3(x, out_planes, stride=1, name=None):
      x = tf.keras.layers.ZeroPadding2D(padding=1, name=f"{name}_pad")(
          x
      )
      return tf.keras.layers.Conv2D(
          filters=out_planes,
          kernel_size=3,
          strides=stride,
          use_bias=False,
          kernel_initializer=_kaiming_normal,
          name=name,
      )(x)
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax import numpy
  import neural_tangents
  from neural_tangents import experimental
  import numpy
  import tensorflow
label_desc: Use image node_id 9 for calling a global function with example usage: _conv3x3(x, out_planes, stride, name) and returns: tf