code: def get_architecture_size(tower_name="search_generator_0"):
      """Returns the number of blocks of the architecture or None if ensembling."""
      prefix = "architectures/{}".format(tower_name)
      architecture = [
          var
          for var in tf.compat.v1.global_variables()
          if prefix in var.name
      ]
      assert len(architecture) <= 1
      size = None
      if architecture:
          size = architecture[0].shape[0]
      return size
  
imports: import collections
  import os
  import string
  import time
  from absl import flags
  from absl import logging
  from model_search import block
  from model_search import block_builder
  from model_search import hparam
  from model_search import utils
  from model_search.proto import hparam_pb2
  from model_search.proto import phoenix_spec_pb2
  from model_search.proto import transfer_learning_spec_pb2
  import numpy
  import tensorflow.compat.v2
  import tf_slim
  from google.protobuf import text_format
label_desc: Use image node_id 7 for calling a global function with example usage: get_architecture_size(tower_name) and returns: size