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

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.