ReplicateMetadata

ReplicateMetadata public final class

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

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

Clases anidadas

clase

ReplicateMetadata.Options

Los atributos opcionales para ReplicateMetadata

Constantes

Cuerda

OP_NAME

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

Métodos públicos

estáticas ReplicateMetadata.Options	allowSoftPlacement (Boolean allowSoftPlacement)
estáticas ReplicateMetadata.Options	computationShape (List <Larga> computationShape)
estática ReplicateMetadata	crear ( Alcance alcance, largas numReplicas, Opciones ... Opciones) Método de fábrica para crear una clase que envuelva una nueva operación ReplicateMetadata.
estáticas ReplicateMetadata.Options	deviceAssignment (List <Larga> deviceAssignment)
estáticas ReplicateMetadata.Options	hostComputeCore (List <String> hostComputeCore)
estáticas ReplicateMetadata.Options	numCoresPerReplica (Long numCoresPerReplica)
estáticas ReplicateMetadata.Options	paddingMap (List <String> paddingMap)
estáticas ReplicateMetadata.Options	stepMarkerLocation (String stepMarkerLocation)
estáticas ReplicateMetadata.Options	topología (topología String)
estáticas ReplicateMetadata.Options	useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)
estáticas ReplicateMetadata.Options	useTpu (Boolean useTpu)

Métodos heredados

De la clase org.tensorflow.op.RawOp

booleano final	es igual a (Object obj)
int final	hashCode ()
Operación	op () Devuelva esta unidad de cálculo como una sola Operation .
cadena final	toString ()

De la clase java.lang.Object

booleano	equals (Object arg0)
Clase final <?>	getClass ()
En t	hashCode ()
vacío final	notificar ()
vacío final	notifyAll ()
Cuerda	toString ()
vacío final	espera (tiempo arg0, arg1 int)
vacío final	espera (arg0 largo)
vacío final	wait ()

Desde la interfaz org.tensorflow.op.Op

abstracta ExecutionEnvironment	env () Devuelve el entorno de ejecución en el que se creó esta operación.
abstracta Operación	op () Devuelva esta unidad de cálculo como una sola Operation .

Constantes

OP_NAME pública final static String

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

Valor constante: "TPUReplicateMetadata"

Métodos públicos

public static ReplicateMetadata.Options allowSoftPlacement (booleano allowSoftPlacement)

public static ReplicateMetadata.Options computationShape (List <Larga> computationShape)

Parámetros

computationShape	OBSOLETO. Utilice num_cores_per_replica en su lugar.

estáticas pública ReplicateMetadata crean ( Alcance alcance, Long numReplicas, Opciones ... Opciones)

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

Parámetros

alcance	alcance actual
numReplicas	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 <Larga> deviceAssignment)

Parámetros

deviceAssignment	La asignación de dispositivos para el cálculo.

estáticas pública ReplicateMetadata.Options hostComputeCore (List <String> hostComputeCore)

public static ReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

Parámetros

numCoresPerReplica	Número de núcleos por réplica. Se utiliza para el paralelismo de modelos.

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

public static ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

estáticas pública ReplicateMetadata.Options topología (topología de cadena)

Parámetros

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

estáticas pública ReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

public static ReplicateMetadata.Options useTpu (booleano useTpu)

Parámetros

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