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