ConfigureDistributedTPU

public final class ConfigureDistributedTPU

Sets up the centralized structures for a distributed TPU system.

Nested Classes

class

ConfigureDistributedTPU.Options

Optional attributes for ConfigureDistributedTPU

Constants

String

OP_NAME

The name of this op, as known by TensorFlow core engine

Public Methods

Output<TString>	asOutput() Returns the symbolic handle of the tensor.
static ConfigureDistributedTPU.Options	compilationFailureClosesChips(Boolean compilationFailureClosesChips)
static ConfigureDistributedTPU	create(Scope scope, Options... options) Factory method to create a class wrapping a new ConfigureDistributedTPU operation.
static ConfigureDistributedTPU.Options	embeddingConfig(String embeddingConfig)
static ConfigureDistributedTPU.Options	enableWholeMeshCompilations(Boolean enableWholeMeshCompilations)
static ConfigureDistributedTPU.Options	isGlobalInit(Boolean isGlobalInit)
Output<TString>	topology() A serialized tensorflow.tpu.TopologyProto that describes the TPU topology.
static ConfigureDistributedTPU.Options	tpuEmbeddingConfig(String tpuEmbeddingConfig)

Inherited Methods

From class org.tensorflow.op.RawOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Return this unit of computation as a single Operation.
final String	toString()

From class java.lang.Object

boolean	equals(Object arg0)
final Class<?>	getClass()
int	hashCode()
final void	notify()
final void	notifyAll()
String	toString()
final void	wait(long arg0, int arg1)
final void	wait(long arg0)
final void	wait()

From interface org.tensorflow.op.Op

abstract ExecutionEnvironment	env() Return the execution environment this op was created in.
abstract Operation	op() Return this unit of computation as a single Operation.

From interface org.tensorflow.Operand

abstract Output<TString>	asOutput() Returns the symbolic handle of the tensor.
abstract TString	asTensor() Returns the tensor at this operand.
abstract Shape	shape() Returns the (possibly partially known) shape of the tensor referred to by the Output of this operand.
abstract Class<TString>	type() Returns the tensor type of this operand

From interface org.tensorflow.ndarray.Shaped

abstract int	rank()
abstract Shape	shape()
abstract long	size() Computes and returns the total size of this container, in number of values.

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "ConfigureDistributedTPU"

Public Methods

public Output<TString> asOutput ()

Returns the symbolic handle of the tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

public static ConfigureDistributedTPU.Options compilationFailureClosesChips (Boolean compilationFailureClosesChips)

public static ConfigureDistributedTPU create (Scope scope, Options... options)

Factory method to create a class wrapping a new ConfigureDistributedTPU operation.

Parameters

scope	current scope
options	carries optional attributes values

Returns

• a new instance of ConfigureDistributedTPU

public static ConfigureDistributedTPU.Options embeddingConfig (String embeddingConfig)

Parameters

embeddingConfig	Reserved. Do not use.

public static ConfigureDistributedTPU.Options enableWholeMeshCompilations (Boolean enableWholeMeshCompilations)

public static ConfigureDistributedTPU.Options isGlobalInit (Boolean isGlobalInit)

Parameters

isGlobalInit	Reserved. Do not use.

public Output<TString> topology ()

A serialized tensorflow.tpu.TopologyProto that describes the TPU topology.

public static ConfigureDistributedTPU.Options tpuEmbeddingConfig (String tpuEmbeddingConfig)

Parameters

tpuEmbeddingConfig	Serialized tensorflow.tpu.TPUEmbeddingConfiguration that describes the embedding lookups of the program.