org.tensorflow

Class Graph

java.lang.Object

org.tensorflow.Graph

All Implemented Interfaces:

AutoCloseable

public final class Graph
extends Object
implements AutoCloseable

A data flow graph representing a TensorFlow computation.

Instances of a Graph are thread-safe.

WARNING: Resources consumed by the Graph object must be explicitly freed by invoking the close() method then the Graph object is no longer needed.

Constructor Summary

Constructors

Constructor and Description
Graph() Create an empty Graph.

Method Summary

Modifier and Type	Method and Description
void	close() Release resources associated with the Graph.
void	importGraphDef(byte[] graphDef) Import a serialized representation of a TensorFlow graph.
void	importGraphDef(byte[] graphDef, String prefix) Import a serialized representation of a TensorFlow graph.
OperationBuilder	opBuilder(String type, String name) Returns a builder to add Operations to the Graph.
Operation	operation(String name) Returns the operation (node in the Graph) with the provided name.
Iterator<Operation>	operations() Iterator over all the Operations in the graph.
byte[]	toGraphDef() Generate a serialized representation of the Graph.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

Graph

public Graph()

Create an empty Graph.

Method Detail

close

public void close()

Release resources associated with the Graph.

Blocks until there are no active Session instances referring to this Graph. A Graph is not usable after close returns.

Specified by:

close in interface AutoCloseable

operation

public Operation operation(String name)

Returns the operation (node in the Graph) with the provided name.

Or null if no such operation exists in the Graph.

operations

public Iterator<Operation> operations()

Iterator over all the Operations in the graph.

The order of iteration is unspecified. Consumers of the iterator will receive no notification should the underlying graph change during iteration.

opBuilder

public OperationBuilder opBuilder(String type,
                                  String name)

Returns a builder to add Operations to the Graph.

Parameters:

type - of the Operation (i.e., identifies the computation to be performed)

name - to refer to the created Operation in the graph.

Returns:

an OperationBuilder, which will add the Operation to the graph when OperationBuilder.build() is invoked. If OperationBuilder.build() is not invoked, then some resources may leak.

importGraphDef

public void importGraphDef(byte[] graphDef)
                    throws IllegalArgumentException

Import a serialized representation of a TensorFlow graph.

The serialized representation of the graph, often referred to as a GraphDef, can be generated by toGraphDef() and equivalents in other language APIs.

Throws:

IllegalArgumentException - if graphDef is not a recognized serialization of a graph.

See Also:

importGraphDef(byte[], String)

importGraphDef

public void importGraphDef(byte[] graphDef,
                           String prefix)
                    throws IllegalArgumentException

Import a serialized representation of a TensorFlow graph.

Parameters:

graphDef - the serialized representation of a TensorFlow graph.

prefix - a prefix that will be prepended to names in graphDef

Throws:

IllegalArgumentException - if graphDef is not a recognized serialization of a graph.

See Also:

importGraphDef(byte[])

toGraphDef

public byte[] toGraphDef()

Generate a serialized representation of the Graph.

See Also:

importGraphDef(byte[]), importGraphDef(byte[], String)