TPUReplicateMetadata.Options

classe estática pública TPUReplicateMetadata.Options

Atributos opcionais para TPUReplicateMetadata

Métodos Públicos

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

Métodos herdados

Métodos Públicos

public TPUReplicateMetadata.Options permitirSoftPlacement (booleano permitirSoftPlacement)

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

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

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

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

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

público TPUReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica longo)

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

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

public TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

topologia pública TPUReplicateMetadata.Options (topologia String)

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

public TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

public TPUReplicateMetadata.Options useTpu (booleano useTpu)

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