ReplicateMetadata

classe finale pubblica ReplicateMetadata

Metadati che indicano come replicare il calcolo della TPU.

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

Classi nidificate

classe ReplicateMetadata.Options Attributi facoltativi per ReplicateMetadata

Costanti

Corda OP_NAME Il nome di questa operazione, come noto al motore principale di TensorFlow

Metodi pubblici

ReplicateMetadata.Options statico
allowSoftPlacement (allowSoftPlacement booleano)
ReplicateMetadata.Options statico
computationShape (Lista<Long> computationShape)
ReplicateMetadata statico
crea (ambito ambito , numRepliche lunghe, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione ReplicateMetadata.
ReplicateMetadata.Options statico
assegnazione dispositivo (Elenco<lungo> assegnazione dispositivo)
ReplicateMetadata.Options statico
hostComputeCore (Lista<String> hostComputeCore)
ReplicateMetadata.Options statico
numCoresPerReplica (numCoresPerReplica lungo)
ReplicateMetadata.Options statico
imbottituraMap (Lista<String> imbottituraMap)
ReplicateMetadata.Options statico
stepMarkerLocation (Stringa stepMarkerLocation)
ReplicateMetadata.Options statico
topologia (topologia di stringa)
ReplicateMetadata.Options statico
useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)
ReplicateMetadata.Options statico
useTpu (useTpu booleano)

Metodi ereditati

Costanti

Stringa finale statica pubblica OP_NAME

Il nome di questa operazione, come noto al motore principale di TensorFlow

Valore costante: "TPUReplicateMetadata"

Metodi pubblici

pubblico statico ReplicateMetadata.OptionsallowSoftPlacement ( allowSoftPlacement booleano)

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

Parametri
calcoloShape DEPRECATO. Utilizzare invece num_cores_per_replica.

creazione pubblica statica ReplicateMetadata (ambito ambito , numReplicas lunghi, opzioni... opzioni)

Metodo factory per creare una classe che racchiude una nuova operazione ReplicateMetadata.

Parametri
scopo ambito attuale
numRepliche Numero di repliche del calcolo
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di ReplicateMetadata

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

Parametri
dispositivoAssegnazione L'assegnazione dei dispositivi per il calcolo.

pubblico statico ReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

pubblico statico ReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica lungo)

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

pubblico statico ReplicateMetadata.Options riempimentoMap (List<String> riempimentoMap)

public static ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

Topologia statica pubblica ReplicateMetadata.Options (topologia String)

Parametri
topologia TopologyProto che indica la topologia della sezione del pod in TPU.

public static ReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

pubblico statico ReplicateMetadata.Options useTpu (useTpu booleano)

Parametri
utilizzareTpu Se posizionare il calcolo sul TPU.