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

De la clase org.tensorflow.op.RawOp

booleano final	es igual (Objeto obj)
entero final	código hash ()
Operación	operación () Devuelve esta unidad de cálculo como una única Operation .
cadena final	Encadenar ()

De la clase java.lang.Object

booleano	es igual (Objeto arg0)
Clase final<?>	obtenerclase ()
En t	código hash ()
vacío final	notificar ()
vacío final	notificar a todos ()
Cadena	Encadenar ()
vacío final	esperar (arg0 largo, int arg1)
vacío final	espera (largo arg0)
vacío final	esperar ()

Desde la interfaz org.tensorflow.op.Op

entorno de ejecución abstracto	entorno () Devuelve el entorno de ejecución en el que se creó esta operación.
operación abstracta	operación () Devuelve esta unidad de cálculo como una única Operation .

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.