Advertencia: esta API está obsoleta y se eliminará en una versión futura de TensorFlow una vez que el reemplazo sea ​​estable.

TPUReplicateMetadata

Organiza tus páginas con colecciones Guarda y categoriza el contenido según tus preferencias.
clase final pública TPUReplicateMetadata

Metadatos que indican cómo se debe replicar el cálculo de TPU.

Esta operación contiene los metadatos comunes a las operaciones de un subgráfico de cálculo `tpu.replicate()`.

Clases anidadas

clase TPUReplicateMetadata.Opciones Atributos opcionales para TPUReplicateMetadata

Métodos públicos

TPUReplicateMetadata.Opciones estáticas
allowSoftPlacement (booleano allowSoftPlacement)
TPUReplicateMetadata.Opciones estáticas
computationShape (List<Long> computationShape)
TPUReplicateMetadata estático
crear ( Ámbito de ámbito, Long numReplicas, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación TPUReplicateMetadata.
TPUReplicateMetadata.Opciones estáticas
asignación de dispositivo (lista <larga> asignación de dispositivo)
TPUReplicateMetadata.Opciones estáticas
hostComputeCore (Lista<String> hostComputeCore)
TPUReplicateMetadata.Opciones estáticas
numCoresPerReplica (Número largo de núcleos por réplica)
TPUReplicateMetadata.Opciones estáticas
paddingMap (Lista<String> paddingMap)
TPUReplicateMetadata.Opciones estáticas
stepMarkerLocation (Cadena stepMarkerLocation)
TPUReplicateMetadata.Opciones estáticas
topología (topología de cadena)
TPUReplicateMetadata.Opciones estáticas
tpuCompileOptionsProto (Cadena tpuCompileOptionsProto)
TPUReplicateMetadata.Opciones estáticas
useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)
TPUReplicateMetadata.Opciones estáticas
useTpu (booleano useTpu)

Métodos Heredados

Métodos públicos

público estático TPUReplicateMetadata.Options allowSoftPlacement (booleano allowSoftPlacement)

public static TPUReplicateMetadata.Options computationShape (List<Long> computationShape)

Parámetros
ComputaciónForma OBSOLETO. Utilice num_cores_per_replica en su lugar.

public static TPUReplicateMetadata create ( Scope scope, Long numReplicas, Options... options)

Método de fábrica para crear una clase que envuelve una nueva operación TPUReplicateMetadata.

Parámetros
alcance alcance actual
número de réplicas Número de réplicas del cálculo
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de TPUReplicateMetadata

pública estática TPUReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)

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

public static TPUReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

public static TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

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

public static TPUReplicateMetadata.Options paddingMap (List<String> paddingMap)

public static TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

topología pública estática TPUReplicateMetadata.Options ( topología de cadena)

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

público estático TPUReplicateMetadata.Options tpuCompileOptionsProto (String tpuCompileOptionsProto)

public static TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

public static TPUReplicateMetadata.Options useTpu (booleano useTpu)

Parámetros
usoTpu Si colocar el cálculo en la TPU.