code: def _make_transition(data, num_output_features, index):
      """Transition between layers."""
      bn = layers.batch_norm_infer(data, name=f"batch_t_{index}")
      relu = relay.nn.relu(bn)
      conv = layers.conv2d(
          relu,
          channels=num_output_features,
          kernel_size=(1, 1),
          name=f"conv_t_{index}",
      )
      return relay.nn.avg_pool2d(conv, pool_size=(2, 2), strides=(2, 2))
  
file_path: tvm/python/tvm/relay/testing/densenet.py
imports: from tvm import relay
  from .None import layers
  from .init import create_workload
label_desc: Use image node_id 3 for calling a global function with example usage: _make_transition(data, num_output_features, index) and returns: relay