code: def _conv2d_normal_params_model_fn(
      out_channels, size, in_channels, dtype
  ):
      yield Root(
          sample.Sample(
              normal.Normal(tf.zeros([], dtype), 1.0),
              list(size) + [in_channels, out_channels],
              name="kernel",
          )
      )
  
imports: import collections
  import functools
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import deterministic
  from tensorflow_probability.python.distributions import joint_distribution_coroutine
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.distributions import sample
  from tensorflow_probability.python.internal import vectorization_util
label_desc: Use image node_id 3 for calling a global function with example usage: _conv2d_normal_params_model_fn(out_channels, size, in_channels, dtype) without return types