code: def _get_blob_out_degree(spec):
      """
      Computes use count of every tensor/node in NN graph
      i.e. How many layers are using it as an input
  
      :param nn_spec : NeuralNetworkSpecification
      :returns use_count_dict : str -> int, a dictionary with node name as a key and it's use count as a value
      """
  
      def _get_blob_out_degree_rec(nn_spec, out_degree):
          nn_layers = nn_spec.layers
          for layer in nn_layers:
              layer_type = layer.WhichOneof("layer")
              for inp in layer.input:
                  out_degree[inp] = out_degree.get(inp, 0) + 1
              if layer_type == "loop":
                  out_degree[layer.loop.conditionVar] = (
                      out_degree.get(layer.loop.conditionVar, 0) + 1
                  )
                  _get_blob_out_degree_rec(
                      layer.loop.conditionNetwork, out_degree
                  )
                  _get_blob_out_degree_rec(
                      layer.loop.bodyNetwork, out_degree
                  )
              elif layer_type == "branch":
                  _get_blob_out_degree_rec(
                      layer.branch.ifBranch, out_degree
                  )
                  _get_blob_out_degree_rec(
                      layer.branch.elseBranch, out_degree
                  )
  
      use_count_dict = {}
      # Collect variable use count recursively
      nn_spec = _get_nn_spec(spec)
      _get_blob_out_degree_rec(nn_spec, use_count_dict)
  
      # Network outputs are variable use
      network_outputs = _get_network_output(spec)
      for _output in network_outputs:
          use_count_dict[_output] = use_count_dict.get(_output, 0) + 1
      return use_count_dict
  
file_path: turicreate/src/external/coremltools_wrap/coremltools/coremltools/converters/mil/backend/nn/passes/mlmodel_passes.py
imports: from __future__ import print_function
  from __future__ import division
  from __future__ import absolute_import
  from itertools import permutations
label_desc: Use image node_id 2 for calling a global function with example usage: _get_blob_out_degree(spec) and returns: use_count_dict