TPUReplicateMetadata

classe finale pubblica TPUReplicateMetadata

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 TPUReplicateMetadata.Options Attributi facoltativi per TPUReplicateMetadata

Costanti

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

Metodi pubblici

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

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

Parametri
calcoloShape DEPRECATO. Utilizzare invece num_cores_per_replica.

creazione pubblica statica TPUReplicateMetadata (ambito ambito , numReplicas lunghi, Opzioni... opzioni)

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

Parametri
scopo ambito attuale
numRepliche 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
dispositivoAssegnazione L'assegnazione dei dispositivi per il calcolo.

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

pubblico statico TPUReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica lungo)

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

pubblico statico TPUReplicateMetadata.Options imbottituraMap (List<String> imbottituraMap)

pubblico statico TPUReplicateMetadata.Options stepMarkerLocation (Stringa stepMarkerLocation)

Topologia statica pubblica TPUReplicateMetadata.Options (topologia stringa)

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

pubblico statico TPUReplicateMetadata.Options useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)

pubblico statico TPUReplicateMetadata.Options useTpu (useTpu booleano)

Parametri
utilizzareTpu Se posizionare il calcolo sul TPU.