tf.compat.v1.GraphDef | TensorFlow v2.16.1 (original) (raw)
tf.compat.v1.GraphDef
Stay organized with collections Save and categorize content based on your preferences.
A protobuf containing the graph of operations.
Migrate to TF2
This API is not available in TensorFlow 2.x.
You should not need to use GraphDefs directly in TF2. To load GraphDefs in TF2, use SavedModel. The SavedModel contains the GraphDef.
Before:
with tf.io.gfile.GFile('/tmp/graph.pb', 'rb') as f:
graph_def = tf.compat.v1.GraphDef()
graph_def.ParseFromString(f.read())
After:
tf.saved_model.load('/tmp/saved_model')
If you would like to create a GraphDef in TF2, use tf.function andget_concrete_function.
@tf.function
def f(x):
return x
``
graph_def = f.get_concrete_function(1.).graph.as_graph_def()
print(graph_def)
Description
Used in the notebooks
| Used in the guide |
|---|
| Migrating your TFLite code to TF2 |
| Attributes | |
|---|---|
| debug_info | GraphDebugInfo debug_info |
| library | FunctionDefLibrary library |
| node | repeated NodeDef node |
| version | int32 version |
| versions | VersionDef versions |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-04-26 UTC.