Add new TensorFlow opeation to existing (freezed) graph

add_tf_operation_to_graph.py

import tensorflow as tf

def get_graph_def_from_file(graph_filepath):
  with tf.Graph().as_default():
    with tf.gfile.GFile(graph_filepath, 'rb') as f:
      graph_def = tf.GraphDef()
      graph_def.ParseFromString(f.read())
      return graph_def

# inputs
graph_file="optimized_tRT.pb"
graph_file_out="optimized_tRT_modified.pb"
outputs=["test_model/model/logits/linear/BiasAdd:0"]

g=tf.Graph()
with tf.Session(graph=g) as sess:
        y, = tf.import_graph_def(
            get_graph_def_from_file(graph_file),
            name="",
            return_elements=outputs
        )
        # Add desired operation
        # Examples:
        #y = tf.argmax(y, axis=3, name="argmax_1", output_type=tf.dtypes.int32)
        y=tf.nn.softmax(y, axis=3, name="softmax_1")
        # Save graph
        tf.train.write_graph(g, logdir=".", as_text=False, name=graph_file_out)