public static class TPUReplicateMetadata.Options extends Object
TPUReplicateMetadata
Modifier and Type | Method and Description |
---|---|
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 |
useTpu(Boolean useTpu) |
public TPUReplicateMetadata.Options numCoresPerReplica(Long numCoresPerReplica)
numCoresPerReplica
- Number of cores per replica. Used for model parallelism.public TPUReplicateMetadata.Options topology(String topology)
topology
- TopologyProto indicating the topology of the TPU pod slice.public TPUReplicateMetadata.Options useTpu(Boolean useTpu)
useTpu
- Whether to place the computation on the TPU.public TPUReplicateMetadata.Options deviceAssignment(List<Long> deviceAssignment)
deviceAssignment
- The assignment of devices for the computation.public TPUReplicateMetadata.Options computationShape(List<Long> computationShape)
computationShape
- DEPRECATED. Use num_cores_per_replica instead.public TPUReplicateMetadata.Options hostComputeCore(List<String> hostComputeCore)
hostComputeCore
- public TPUReplicateMetadata.Options paddingMap(List<String> paddingMap)
paddingMap
- public TPUReplicateMetadata.Options stepMarkerLocation(String stepMarkerLocation)
stepMarkerLocation
- public TPUReplicateMetadata.Options allowSoftPlacement(Boolean allowSoftPlacement)
allowSoftPlacement
- Copyright © 2022. All rights reserved.