TPUReplicateMetadata.Options

classe statica pubblica TPUReplicateMetadata.Options

Attributi facoltativi per TPUReplicateMetadata

Metodi pubblici

TPUReplicateMetadata.Options
allowSoftPlacement (allowSoftPlacement booleano)
TPUReplicateMetadata.Options
computationShape (Lista<Long> computationShape)
TPUReplicateMetadata.Options
assegnazione dispositivo (Elenco<lungo> assegnazione dispositivo)
TPUReplicateMetadata.Options
hostComputeCore (Lista<String> hostComputeCore)
TPUReplicateMetadata.Options
numCoresPerReplica (numCoresPerReplica lungo)
TPUReplicateMetadata.Options
imbottituraMap (Lista<String> imbottituraMap)
TPUReplicateMetadata.Options
stepMarkerLocation (Stringa stepMarkerLocation)
TPUReplicateMetadata.Options
topologia (topologia di stringa)
TPUReplicateMetadata.Options
useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)
TPUReplicateMetadata.Options
useTpu (useTpu booleano)

Metodi ereditati

Metodi pubblici

public TPUReplicateMetadata.OptionsallowSoftPlacement ( allowSoftPlacement booleano)

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

Parametri
calcoloShape DEPRECATO. Utilizzare invece num_cores_per_replica.

public TPUReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)

Parametri
dispositivoAssegnazione L'assegnazione dei dispositivi per il calcolo.

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

public TPUReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica lungo)

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

public TPUReplicateMetadata.Options imbottituraMap (List<String> imbottituraMap)

public TPUReplicateMetadata.Options stepMarkerLocation (Stringa stepMarkerLocation)

topologia pubblica TPUReplicateMetadata.Options (topologia stringa)

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

public TPUReplicateMetadata.Options useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)

public TPUReplicateMetadata.Options useTpu (useTpu booleano)

Parametri
utilizzareTpu Se posizionare il calcolo sul TPU.