ReplicateMetadata.Options

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.OptionsallowSoftPlacement ( allowSoftPlacement booleano)

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.

public ReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

pubblico ReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica lungo)

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

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

pubblico ReplicateMetadata.Options stepMarkerLocation (Stringa stepMarkerLocation)

topologia pubblica ReplicateMetadata.Options (topologia stringa)

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

public ReplicateMetadata.Options useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)

public ReplicateMetadata.Options useTpu (useTpu booleano)

Parametri
utilizzareTpu Se posizionare il calcolo sul TPU.