How to "append" Op at the beginning of a TensorFlow graph?

Question

I have a GraphDef proto file which I am importing using tf.import_graph_def. Ops can be added at the end of the graph like this:

final_tensor = tf.import_graph_def(graph_def, name='', return_elements=['final_tensor'])
new_tensor = some_op(final_tensor)

But I want to add Ops at the beginning of the graph, so essentially the first Op in the graph_def needs to take the output of my Op as input, how do I do it?

Answer 1

Finally found a way to do this. I am sure the function Yarolsav mentioned in the comments does something similar internally.

new_input = graph_def.node.add()
new_input.op = 'new_op_name'  # eg: 'Const', 'Placeholder', 'Add' etc
new_input.name = 'some_new_name'
# set any attributes you want for new_input here
old_input.input[0] = 'some_new_name'  #  must match with the name above

For details about how to set the attributes, see this file.