Sets up the centralized structures for a distributed TPU system.
Nested Classes
class | ConfigureDistributedTPU.Options |
Optional attributes for
ConfigureDistributedTPU
|
Public Methods
Output <String> |
asOutput
()
Returns the symbolic handle of a 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 <String> |
topology
()
A serialized tensorflow.tpu.TopologyProto that describes the TPU
topology.
|
static ConfigureDistributedTPU.Options |
tpuEmbeddingConfig
(String tpuEmbeddingConfig)
|
Inherited Methods
Public Methods
public Output <String> asOutput ()
Returns the symbolic handle of a 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 <String> 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. |
---|