ReplicateMetadata.Options

classe estática pública ReplicateMetadata.Options

Atributos opcionais para ReplicateMetadata

Métodos Públicos

ReplicateMetadata.Options
permitirSoftPlacement (booleano permitirSoftPlacement)
ReplicateMetadata.Options
computationShape (List<Long> computationShape)
ReplicateMetadata.Options
deviceAssignment (Lista<Long> deviceAssignment)
ReplicateMetadata.Options
hostComputeCore (List<String> hostComputeCore)
ReplicateMetadata.Options
numCoresPerReplica (numCoresPerReplica longo)
ReplicateMetadata.Options
paddingMap (List<String> paddingMap)
ReplicateMetadata.Options
stepMarkerLocation (String stepMarkerLocation)
ReplicateMetadata.Options
topologia (topologia de string)
ReplicateMetadata.Options
useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)
ReplicateMetadata.Options
useTpu (booleano useTpu)

Métodos herdados

Métodos Públicos

public ReplicateMetadata.Options permitirSoftPlacement (booleano permitirSoftPlacement)

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

Parâmetros
computaçãoForma DESCONTINUADA. Use num_cores_per_replica em vez disso.

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

Parâmetros
atribuição de dispositivo A atribuição de dispositivos para a computação.

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

público ReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica longo)

Parâmetros
numCoresPerReplica Número de núcleos por réplica. Usado para paralelismo de modelo.

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

public ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

Topologia pública ReplicateMetadata.Options (topologia String)

Parâmetros
topologia TopologyProto indicando a topologia da fatia do pod TPU.

public ReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

public ReplicateMetadata.Options useTpu (booleano useTpu)

Parâmetros
usarTpu Se o cálculo deve ser colocado na TPU.