code: def _make_pixel_cnn(self, use_weight_norm_and_data_init=False):
      return pixel_cnn.PixelCNN(
          image_shape=self.image_shape,
          conditional_shape=self.h_shape,
          num_resnet=2,
          num_hierarchies=2,
          num_filters=1,
          num_logistic_mix=self.num_logistic_mix,
          high=self.high,
          low=self.low,
          use_weight_norm=use_weight_norm_and_data_init,
          use_data_init=use_weight_norm_and_data_init,
      )
  
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import pixel_cnn
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.internal import tf_keras
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 2 for calling the PixelCnnTest obj's underlying member method code with example usage: obj._make_pixel_cnn(use_weight_norm_and_data_init) and returns: pixel_cnn