ReplicateMetadata.Options

clase estática pública ReplicateMetadata.Options

Atributos opcionales para ReplicateMetadata

Métodos públicos

ReplicarMetadatos.Opciones
permitirSoftPlacement (permiso booleanoSoftPlacement)
ReplicarMetadatos.Opciones
forma de cálculo (Lista<Long> forma de cálculo)
ReplicarMetadatos.Opciones
Asignación de dispositivo (Lista<Long> asignación de dispositivo)
ReplicarMetadatos.Opciones
hostComputeCore (Lista<Cadena> hostComputeCore)
ReplicarMetadatos.Opciones
numCoresPerReplica (numCoresPerReplica largo)
ReplicarMetadatos.Opciones
paddingMap (Lista<String> paddingMap)
ReplicarMetadatos.Opciones
stepMarkerLocation (cadena stepMarkerLocation)
ReplicarMetadatos.Opciones
topología (topología de cadena)
ReplicarMetadatos.Opciones
useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)
ReplicarMetadatos.Opciones
useTpu (uso booleano Tpu)

Métodos heredados

Métodos públicos

public ReplicateMetadata.Options enableSoftPlacement (permiso booleanoSoftPlacement)

public ReplicateMetadata.Options computationShape (List<Long> computationShape)

Parámetros
cálculoForma OBSOLETO. Utilice num_cores_per_replica en su lugar.

public ReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)

Parámetros
dispositivoAsignación La asignación de dispositivos para el cómputo.

public ReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

public ReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica largo)

Parámetros
número de núcleos por réplica Número de núcleos por réplica. Se utiliza para el paralelismo de modelos.

público ReplicateMetadata.Options paddingMap (List<String> paddingMap)

público ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

Topología pública ReplicateMetadata.Options (topología de cadena)

Parámetros
topología TopologyProto que indica la topología del segmento del pod de TPU.

public ReplicateMetadata.Options useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)

public ReplicateMetadata.Options useTpu (useTpu booleano)

Parámetros
utilizarTpu Ya sea para colocar el cálculo en la TPU.