TPUReplicateMetadata

public final class TPUReplicateMetadata

Metadati che indicano come replicare il calcolo TPU.

Questa operazione contiene i metadati comuni alle operazioni di un sottografo di calcolo `tpu.replicate ()`.

Classi annidate

classe

TPUReplicateMetadata.Options

Attributi facoltativi per TPUReplicateMetadata

Metodi pubblici

static TPUReplicateMetadata.Options	allowSoftPlacement (Boolean allowSoftPlacement)
TPUReplicateMetadata.Options statico	computationShape (List <Long> computationShape)
TPUReplicateMetadata statico	create ( Scope scope, Long numReplicas, Options ... options) Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione TPUReplicateMetadata.
static TPUReplicateMetadata.Options	deviceAssignment (List <Long> deviceAssignment)
static TPUReplicateMetadata.Options	hostComputeCore (List <String> hostComputeCore)
static TPUReplicateMetadata.Options	numCoresPerReplica (Long numCoresPerReplica)
static TPUReplicateMetadata.Options	paddingMap (List <String> paddingMap)
static TPUReplicateMetadata.Options	stepMarkerLocation (String stepMarkerLocation)
static TPUReplicateMetadata.Options	topologia ( topologia di stringa)
TPUReplicateMetadata.Options statico	useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)
TPUReplicateMetadata.Options statico	useTpu (booleano useTpu)

Metodi ereditati

Dalla classe org.tensorflow.op.PrimitiveOp

booleano finale	è uguale a (Object obj)
final int	hashCode ()
Operazione	op () Restituisce l' Operation sottostante
stringa finale	toString ()

Dalla classe java.lang.Object

booleano	è uguale a (Oggetto arg0)
classe finale <?>	getClass ()
int	hashCode ()
vuoto finale	notifica ()
vuoto finale	notifyAll ()
Corda	toString ()
vuoto finale	wait (long arg0, int arg1)
vuoto finale	aspetta (lungo arg0)
vuoto finale	aspetta ()

Metodi pubblici

public static TPUReplicateMetadata.Options allowSoftPlacement (Boolean allowSoftPlacement)

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

Parametri

computationShape	DEPRECATO. Usa invece num_cores_per_replica.

public static TPUReplicateMetadata create ( Scope scope, Long numReplicas, Options ... options)

Metodo Factory per creare una classe che avvolge una nuova operazione TPUReplicateMetadata.

Parametri

scopo	ambito attuale
numReplicas	Numero di repliche del calcolo
opzioni	trasporta valori di attributi opzionali

ritorna

• una nuova istanza di TPUReplicateMetadata

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

Parametri

deviceAssignment	L'assegnazione di dispositivi per il calcolo.

public static TPUReplicateMetadata.Options hostComputeCore (List <String> hostComputeCore)

public static TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

Parametri

numCoresPerReplica	Numero di core per replica. Utilizzato per il parallelismo del modello.

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

public static TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

public static TPUReplicateMetadata.Options topology (String topology)

Parametri

topologia	TopologyProto che indica la topologia dello slice del pod TPU.

public static TPUReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

public static TPUReplicateMetadata.Options useTpu (Boolean useTpu)

Parametri

useTpu	Se posizionare il calcolo sul TPU.