@av8ramit do you know why the TransformGraph class is not in the pip package anymore? The class is defined here.

Thanks for filing the issue @DEKHTIARJonathan. Taking a look and will get back to you.

So I believe that API was removed from the 2.0 pip package. It appears the recommended alternative is to build it from source using bazel. @annarev do you perhaps have more context into alternatives or why it was removed from the 2.0 APIs?

@av8ramit Building from source might be an issue. In principle, I have nothing against it. However in this case we would it in one of the core APIs of Tensorflow.

We would like to use it in the TF-TRT APIs (tensorflow/python/compiler/tensorrt), so we would need to have this available by default in Tensorflow2.

And we can't really ask each TF-TRT user to build TF from source ...

Btw. Kuddos to the team who implemented this API, very nicely put together ;)

Graph transforms were removed and meant to be replaced by Grappler according to this RFC:
https://github.com/tensorflow/community/blob/master/rfcs/20180907-contrib-sunset.md

@petewarden, @rmlarsen do you know if there is an alternative for this usecase (pre-process graph before inference with SavedModel)?

Sorry @DEKHTIARJonathan I am not the best person to answer this (I was looped in in the event this was a pybind export issue), but I'll leave it to Pete and Rasmus to hopefully shed some more light.

@annarev let me give you a crude of what I could be interested in doing if that could help:

transforms = [
        "remove_nodes(op=Identity, op=CheckNumerics, op=StopGradient, op=Placeholder)",
        "sort_by_execution_order",
        "remove_attribute(attribute_name=_XlaSeparateCompiledGradients)",
        "remove_attribute(attribute_name=_XlaCompile)",
        "remove_attribute(attribute_name=_XlaScope)",
        "sort_by_execution_order",
        'merge_duplicate_nodes',
        "sort_by_execution_order",
        'strip_unused_nodes',
        "sort_by_execution_order",
        'fold_constants(ignore_errors=true)',
        "sort_by_execution_order",
        'fold_batch_norms',
        "sort_by_execution_order",
    ]

    print("Applying Graph Transformations ...")
    return TransformGraph(
        graph_def,
        input_node_names,  # an array of the input node(s)
        output_node_names,  # an array of output nodes
        transforms
    )

The order of execution is important. Is there any way to order grappler optimization passes in case I find a way to use the grappler ?

then how to use the transforms in grappler optimization?

Has any progress been made on this? Anybody find a fix?

@av8ramit @reedwm any update ?

@rohan100jain, should these files be added back in the pip package?

@DEKHTIARJonathan I no longer work on this project, unfortunately. Good luck!