classe statica pubblica ReplicateMetadata.Options
Attributi facoltativi per ReplicateMetadata
Metodi pubblici
ReplicateMetadata.Options | allowSoftPlacement (allowSoftPlacement booleano) |
ReplicateMetadata.Options | computationShape (Lista<Long> computationShape) |
ReplicateMetadata.Options | assegnazione dispositivo (Elenco<lungo> assegnazione dispositivo) |
ReplicateMetadata.Options | hostComputeCore (Lista<String> hostComputeCore) |
ReplicateMetadata.Options | numCoresPerReplica (numCoresPerReplica lungo) |
ReplicateMetadata.Options | imbottituraMap (Lista<String> imbottituraMap) |
ReplicateMetadata.Options | stepMarkerLocation (Stringa stepMarkerLocation) |
ReplicateMetadata.Options | topologia (topologia di stringa) |
ReplicateMetadata.Options | useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning) |
ReplicateMetadata.Options | useTpu (useTpu booleano) |
Metodi ereditati
Metodi pubblici
public ReplicateMetadata.Options computationShape (List<Long> computationShape)
Parametri
calcoloShape | DEPRECATO. Utilizzare invece num_cores_per_replica. |
---|
public ReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)
Parametri
dispositivoAssegnazione | L'assegnazione dei dispositivi per il calcolo. |
---|
pubblico ReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica lungo)
Parametri
numCoresPerReplica | Numero di core per replica. Utilizzato per il parallelismo del modello. |
---|
topologia pubblica ReplicateMetadata.Options (topologia stringa)
Parametri
topologia | TopologyProto che indica la topologia della sezione del pod in TPU. |
---|
public ReplicateMetadata.Options useTpu (useTpu booleano)
Parametri
utilizzareTpu | Se posizionare il calcolo sul TPU. |
---|