TPUReplicateMetadata.Options

TPUReplicateMetadata.Options estáticos de clase pública

Los atributos opcionales para TPUReplicateMetadata

Métodos públicos

TPUReplicateMetadata.Options	allowSoftPlacement (Boolean allowSoftPlacement)
TPUReplicateMetadata.Options	computationShape (List <Larga> computationShape)
TPUReplicateMetadata.Options	deviceAssignment (List <Larga> deviceAssignment)
TPUReplicateMetadata.Options	hostComputeCore (List <String> hostComputeCore)
TPUReplicateMetadata.Options	numCoresPerReplica (Long numCoresPerReplica)
TPUReplicateMetadata.Options	paddingMap (List <String> paddingMap)
TPUReplicateMetadata.Options	stepMarkerLocation (String stepMarkerLocation)
TPUReplicateMetadata.Options	topología (topología String)
TPUReplicateMetadata.Options	useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)
TPUReplicateMetadata.Options	useTpu (Boolean useTpu)

Métodos heredados

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

Métodos públicos

pública TPUReplicateMetadata.Options allowSoftPlacement (booleano allowSoftPlacement)

pública TPUReplicateMetadata.Options computationShape (List <Larga> computationShape)

Parámetros

computationShape	OBSOLETO. Utilice num_cores_per_replica en su lugar.

pública TPUReplicateMetadata.Options deviceAssignment (List <Larga> deviceAssignment)

Parámetros

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

pública TPUReplicateMetadata.Options hostComputeCore (List <String> hostComputeCore)

pública TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

Parámetros

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

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

pública TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

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.

pública TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

pública TPUReplicateMetadata.Options useTpu (booleano useTpu)

Parámetros

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