code: def SearchTFProfNode(node, name):
      """Search a node in the tree."""
      if node.name == name:
          return node
      for c in node.children:
          r = SearchTFProfNode(c, name)
          if r:
              return r
      return None
  
imports: import contextlib
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import init_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn_grad
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import rnn
  from tensorflow.python.ops import rnn_cell
  from tensorflow.python.ops import tensor_array_grad
  from tensorflow.python.ops import variable_scope
  from tensorflow.python.profiler import model_analyzer
  from tensorflow.python.training import gradient_descent
  from tensorflow.python.util import _pywrap_tfprof
  from tensorflow.python.util import compat
label_desc: Use image node_id 4 for calling a global function with example usage: SearchTFProfNode(node, name) and returns: None, node, r