message: Do not optimize out Identity nodes if specified as output.
diff: @@ -408,7 +408,8 @@ class GraphRewriter(object):
  for output_node in output_nodes:
  self.quantize_nodes_recursively(output_node)
  elif self.mode == "eightbit":
  - self.set_input_graph(graph_util.remove_training_nodes(self.input_graph))
  + self.set_input_graph(graph_util.remove_training_nodes(
  + self.input_graph, protected_nodes=output_node_names))
  output_nodes = [
  self.nodes_map[output_node_name]
  for output_node_name in output_node_names
  