code: def write_png(filename, image):
      """Writes a PNG image file."""
      image = tf.squeeze(image, 0)
      if image.dtype.is_floating:
          image = tf.round(image)
      if image.dtype != tf.uint8:
          image = tf.saturate_cast(image, tf.uint8)
      string = tf.image.encode_png(image)
      tf.io.write_file(filename, string)
  
imports: import argparse
  import io
  import os
  import sys
  import urllib
  from absl import app
  from absl.flags import argparse_flags
  import numpy
  import tensorflow
  import tensorflow_compression
label_desc: Use image node_id 2 for calling a global function with example usage: write_png(filename, image) without return types