saved_model_aot_compile.py removes unwanted tensors from signature #54296
Labels
comp:runtime
c++ runtime, performance issues (cpu)
stat:awaiting response
Status - Awaiting response from author
TF 2.4
for issues related to TF 2.4
type:bug
Bug
System information
Describe the current behavior
When I load the frozen graph using
tf.saved_model.load("saved_model.pb")
, the signature shows all the tensor names, but when.local/lib/python3.6/site-packages/tensorflow/python/tools/saved_model_aot_compile.py
does the_prune_removed_feed_nodes(signature_def, graph_def)
, it prunes multiple tensors that it does not find ingraph_dev.node
and as a result my final deployed model is incorrect:Here are my other TF related packages:
Also,
2.4.1
was used to create the model:Here is the debug output:
The text was updated successfully, but these errors were encountered: