code: def _get_node_name_to_node_def(
      graph_def: tf.compat.v1.GraphDef,
  ) -> Dict[str, tf.compat.v1.NodeDef]:
      return {node.name: node for node in graph_def.node}
  
imports: import collections
  from typing import Dict, List, Mapping, Set
  import tensorflow
  from tfx.dsl.io import fileio
  from tfx.experimental.distributed_inference.graphdef_experiments.subgraph_partitioning import execution_spec
label_desc: Use image node_id 6 for calling a global function with example usage: _get_node_name_to_node_def(graph_def) and returns: unknown