tf.compat.v1.graph_util.remove_training_nodes