code: def get_tf_dtype(flags_obj):
      if (
          getattr(flags_obj, "fp16_implementation", None)
          == "graph_rewrite"
      ):
          # If the graph_rewrite is used, we build the graph with fp32, and let the
          # graph rewrite change ops to fp16.
          return tf.float32
      return DTYPE_MAP[flags_obj.dtype]
  
file_path: autodist/examples/benchmark/utils/flags/_performance.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import multiprocessing
  from absl import flags
  import tensorflow
  from utils.flags._conventions import help_wrap
label_desc: Use image node_id 1 for calling a global function with example usage: get_tf_dtype(flags_obj) and returns: DTYPE_MAP, tf