View source on GitHub |
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 GraphDef
s directly in TF2. To load GraphDef
s 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
and
get_concrete_function
.
@tf.function
def f(x):
return x
graph_def = f.get_concrete_function(1.).graph.as_graph_def()
print(graph_def)
Description
Attributes | |
---|---|
debug_info
|
GraphDebugInfo debug_info
|
library
|
FunctionDefLibrary library
|
node
|
repeated NodeDef node
|
version
|
int32 version
|
versions
|
VersionDef versions
|