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
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 numCoresPerReplica (numCoresPerReplica lungo)
Parametri
numCoresPerReplica | Numero di core per replica. Utilizzato per il parallelismo del modello. |
---|
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. |
---|