code: def set_tensor_final(self, tensor_name):
      
      tensor = self._name_to_tensor(tensor_name)
      self._final_tensors.add(tensor)
docs: Denotes a tensor as a final output of the computation.
  
      Args:
        tensor_name: a string, name of a tensor in the graph.