TPUReplicateMetadata.Options

clase estática pública TPUReplicateMetadata.Options

Atributos opcionales para TPUReplicateMetadata

Métodos públicos

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

Métodos heredados

Métodos públicos

público TPUReplicateMetadata.Options enableSoftPlacement (permiso booleanoSoftPlacement)

pública TPUReplicateMetadata.Options computationShape (List<Long> computationShape)

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

pública TPUReplicateMetadata.Options asignación de dispositivo (Lista<Long> asignación de dispositivo)

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

público TPUReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

público TPUReplicateMetadata.Options numCoresPerReplica (NúmCoresPerReplica 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 TPUReplicateMetadata.Options paddingMap (List<String> paddingMap)

público TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

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

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

público TPUReplicateMetadata.Options tpuCompileOptionsProto (cadena tpuCompileOptionsProto)

público TPUReplicateMetadata.Opciones useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)

público TPUReplicateMetadata.Opciones useTpu (uso booleano Tpu)

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