TPUReplicateMetadata

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 subgrafo de cálculo `tpu.replicate()`.

Clases anidadas

clase Opciones de TPUReplicateMetadata Atributos opcionales para TPUReplicateMetadata

Constantes

Cadena OP_NOMBRE El nombre de esta operación, como lo conoce el motor central de TensorFlow.

Métodos públicos

TPUReplicateMetadata estático.Opciones
permitirSoftPlacement (permiso booleanoSoftPlacement)
TPUReplicateMetadata estático.Opciones
forma de cálculo (Lista<Long> forma de cálculo)
TPUReplicateMetadata estático
crear (alcance alcance , Long numReplicas, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación TPUReplicateMetadata.
TPUReplicateMetadata estático.Opciones
Asignación de dispositivo (Lista<Long> asignación de dispositivo)
TPUReplicateMetadata estático.Opciones
hostComputeCore (Lista<Cadena> hostComputeCore)
TPUReplicateMetadata estático.Opciones
numCoresPerReplica (numCoresPerReplica largo)
TPUReplicateMetadata estático.Opciones
paddingMap (Lista<String> paddingMap)
TPUReplicateMetadata estático.Opciones
stepMarkerLocation (cadena stepMarkerLocation)
TPUReplicateMetadata estático.Opciones
topología (topología de cadena)
TPUReplicateMetadata estático.Opciones
useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)
TPUReplicateMetadata estático.Opciones
useTpu (uso booleano Tpu)

Métodos heredados

Constantes

Cadena final estática pública OP_NAME

El nombre de esta operación, como lo conoce el motor central de TensorFlow.

Valor constante: "TPUReplicateMetadata"

Métodos públicos

TPUReplicateMetadata.Options estático público enableSoftPlacement (permiso booleanoSoftPlacement)

TPUReplicateMetadata.Options estático público computationShape (List<Long> computationShape)

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

Creación pública estática de TPUReplicateMetadata ( alcance , número largo de réplicas, opciones... opciones)

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

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

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

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

TPUReplicateMetadata.Options estático público 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.

TPUReplicateMetadata.Options estático público paddingMap (List<String> paddingMap)

TPUReplicateMetadata.Options estático público 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.

TPUReplicateMetadata.Options estático público useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)

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

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