clase estática pública TPUReplicateMetadata.Options
Atributos opcionales para TPUReplicateMetadata
Métodos públicos
TPUReplicateMetadata.Opciones | allowSoftPlacement (booleano allowSoftPlacement) |
TPUReplicateMetadata.Opciones | computationShape (List<Long> computationShape) |
TPUReplicateMetadata.Opciones | asignación de dispositivo (lista <larga> asignación de dispositivo) |
TPUReplicateMetadata.Opciones | hostComputeCore (Lista<String> hostComputeCore) |
TPUReplicateMetadata.Opciones | numCoresPerReplica (Número largo de núcleos por réplica) |
TPUReplicateMetadata.Opciones | paddingMap (Lista<String> paddingMap) |
TPUReplicateMetadata.Opciones | stepMarkerLocation (Cadena stepMarkerLocation) |
TPUReplicateMetadata.Opciones | topología (topología de cadena) |
TPUReplicateMetadata.Opciones | tpuCompileOptionsProto (Cadena tpuCompileOptionsProto) |
TPUReplicateMetadata.Opciones | useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning) |
TPUReplicateMetadata.Opciones | useTpu (booleano useTpu) |
Métodos Heredados
Métodos públicos
public TPUReplicateMetadata.Options computationShape (List<Long> computationShape)
Parámetros
ComputaciónForma | OBSOLETO. Utilice num_cores_per_replica en su lugar. |
---|
pública TPUReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)
Parámetros
asignación de dispositivo | La asignación de dispositivos para el cómputo. |
---|
public TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
Parámetros
numCoresPerReplica | Número de núcleos por réplica. Se utiliza para el paralelismo del modelo. |
---|
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ública TPUReplicateMetadata.Options useTpu (booleano useTpu)
Parámetros
usoTpu | Si colocar el cálculo en la TPU. |
---|