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. |
---|