org.tensorflow

Class OperationBuilder

java.lang.Object

org.tensorflow.OperationBuilder

public final class OperationBuilder
extends Object

A builder for Operations in a Graph.

Instances of an OperationBuilder are not thread-safe.

A builder for adding Operations to a Graph. For example, the following uses the builder to create an operation that produces the constant "3" as its output:

 // g is a Graph instance.
 try (Tensor c1 = Tensor.create(3.0f)) {
   g.opBuilder("Constant", "MyConst")
       .setAttr("dtype", c1.dataType())
       .setAttr("value", c1)
       .build();
 }
 

Method Summary

Modifier and Type	Method and Description
OperationBuilder	addInput(Output input)
OperationBuilder	addInputList(Output[] inputs)
Operation	build() Add the Operation being built to the Graph.
OperationBuilder	setAttr(String name, boolean value)
OperationBuilder	setAttr(String name, boolean[] value)
OperationBuilder	setAttr(String name, byte[] value)
OperationBuilder	setAttr(String name, DataType value)
OperationBuilder	setAttr(String name, DataType[] value)
OperationBuilder	setAttr(String name, float value)
OperationBuilder	setAttr(String name, float[] value)
OperationBuilder	setAttr(String name, long value)
OperationBuilder	setAttr(String name, long[] value)
OperationBuilder	setAttr(String name, Shape value)
OperationBuilder	setAttr(String name, String value)
OperationBuilder	setAttr(String name, Tensor value)
OperationBuilder	setAttr(String name, Tensor[] value)
OperationBuilder	setDevice(String device)

Methods inherited from class java.lang.Object

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

Method Detail

build

public Operation build()

Add the Operation being built to the Graph.

The OperationBuilder is not usable after build() returns.

addInput

public OperationBuilder addInput(Output input)

addInputList

public OperationBuilder addInputList(Output[] inputs)

setDevice

public OperationBuilder setDevice(String device)

setAttr

public OperationBuilder setAttr(String name,
                                String value)

setAttr

public OperationBuilder setAttr(String name,
                                byte[] value)

setAttr

public OperationBuilder setAttr(String name,
                                long value)

setAttr

public OperationBuilder setAttr(String name,
                                long[] value)

setAttr

public OperationBuilder setAttr(String name,
                                float value)

setAttr

public OperationBuilder setAttr(String name,
                                float[] value)

setAttr

public OperationBuilder setAttr(String name,
                                boolean value)

setAttr

public OperationBuilder setAttr(String name,
                                boolean[] value)

setAttr

public OperationBuilder setAttr(String name,
                                DataType value)

setAttr

public OperationBuilder setAttr(String name,
                                DataType[] value)

setAttr

public OperationBuilder setAttr(String name,
                                Tensor value)

setAttr

public OperationBuilder setAttr(String name,
                                Tensor[] value)

setAttr

public OperationBuilder setAttr(String name,
                                Shape value)