code: def get_node_io_shapes(node, key):
      """Returns the input/output shapes of a GraphDef Node."""
      out_shape = []
      for shape in node.attr[key].list.shape:
          out_shape.append([dim.size for dim in shape.dim])
      return out_shape
  
imports: import collections
  import os
  import re
  from packaging import version
  from tensorflow.compiler.tf2tensorrt import _pywrap_py_utils
  from tensorflow.core.protobuf import rewriter_config_pb2
  from tensorflow.python.framework import dtypes
label_desc: Use image node_id 9 for calling a global function with example usage: get_node_io_shapes(node, key) and returns: out_shape