ReplicateMetadata

clase final pública ReplicateMetadata

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

ReplicarMetadatos.Opciones

Atributos opcionales para ReplicateMetadata

Constantes

Cadena

OP_NOMBRE

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

Métodos públicos

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

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 (arg0 largo)
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

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

pública estática ReplicateMetadata.Options computationShape (List<Long> computationShape)

Parámetros

cálculoForma	OBSOLETO. Utilice num_cores_per_replica en su lugar.

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

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

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 ReplicateMetadata

public static ReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)

Parámetros

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

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

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

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

público estático ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

Topología pública estática ReplicateMetadata.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 ReplicateMetadata.Options useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)

público estático ReplicateMetadata.Options useTpu (useTpu booleano)

Parámetros

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