Hi @shir,
graphsurgeon
allows you to transform TensorFlow graphs. Its capabilities are broadly divided into two categories: search and manipulation. Search functions allow you to find nodes in a TensorFlow graph. Manipulation functions allow you to modify, add, or remove nodes.
You can find more details here
Also please refer to the sample example
Thanks!