public static class
TPUReplicateMetadata.Options
Optional attributes for TPUReplicateMetadata
Public Methods
| TPUReplicateMetadata.Options |
allowSoftPlacement(Boolean allowSoftPlacement)
|
| TPUReplicateMetadata.Options |
computationShape(List<Long> computationShape)
|
| TPUReplicateMetadata.Options |
deviceAssignment(List<Long> deviceAssignment)
|
| TPUReplicateMetadata.Options |
hostComputeCore(List<String> hostComputeCore)
|
| TPUReplicateMetadata.Options |
numCoresPerReplica(Long numCoresPerReplica)
|
| TPUReplicateMetadata.Options |
paddingMap(List<String> paddingMap)
|
| TPUReplicateMetadata.Options |
stepMarkerLocation(String stepMarkerLocation)
|
| TPUReplicateMetadata.Options |
topology(String topology)
|
| TPUReplicateMetadata.Options |
tpuCompileOptionsProto(String tpuCompileOptionsProto)
|
| TPUReplicateMetadata.Options |
useSpmdForXlaPartitioning(Boolean useSpmdForXlaPartitioning)
|
| TPUReplicateMetadata.Options |
useTpu(Boolean useTpu)
|
Inherited Methods
| 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()
|
Public Methods
public TPUReplicateMetadata.Options computationShape (List<Long> computationShape)
Parameters
| computationShape | DEPRECATED. Use num_cores_per_replica instead. |
|---|
public TPUReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)
Parameters
| deviceAssignment | The assignment of devices for the computation. |
|---|
public TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
Parameters
| numCoresPerReplica | Number of cores per replica. Used for model parallelism. |
|---|
public TPUReplicateMetadata.Options topology (String topology)
Parameters
| topology | TopologyProto indicating the topology of the TPU pod slice. |
|---|
public TPUReplicateMetadata.Options useTpu (Boolean useTpu)
Parameters
| useTpu | Whether to place the computation on the TPU. |
|---|