TPUReplicateMetadata

Clase public final TPUReplicateMetadata

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

TPUReplicateMetadata.Options

Los atributos opcionales para TPUReplicateMetadata

Constantes

Cuerda

OP_NAME

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

Métodos públicos

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

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

Parámetros

computationShape	OBSOLETO. Utilice num_cores_per_replica en su lugar.

public static TPUReplicateMetadata crear ( Alcance alcance, Long numReplicas, Opciones ... Opciones)

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

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 TPUReplicateMetadata

public static TPUReplicateMetadata.Options deviceAssignment (List <Larga> deviceAssignment)

Parámetros

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

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 de modelos.

estáticas pública TPUReplicateMetadata.Options paddingMap (List <String> paddingMap)

public static TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

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

Parámetros

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

public static TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

public static TPUReplicateMetadata.Options useTpu (booleano useTpu)

Parámetros

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