tf.io.encode_png