TPUReplicateMetadata.Options

TPUReplicateMetadata.Options classe estática pública

Atributos opcionais para TPUReplicateMetadata

Métodos Públicos

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

Métodos herdados

Métodos Públicos

pública TPUReplicateMetadata.Options allowSoftPlacement (Boolean allowSoftPlacement)

pública TPUReplicateMetadata.Options computationShape (List <longo> computationShape)

Parâmetros
computationShape DESCONTINUADA. Em vez disso, use num_cores_per_replica.

pública TPUReplicateMetadata.Options deviceAssignment (List <longo> deviceAssignment)

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

pública TPUReplicateMetadata.Options hostComputeCore (List <String> hostComputeCore)

pública TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

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

pública TPUReplicateMetadata.Options paddingMap (List <String> paddingMap)

pública TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

pública TPUReplicateMetadata.Options topologia (topologia String)

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

pública TPUReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

pública TPUReplicateMetadata.Options useTpu (Boolean useTpu)

Parâmetros
useTpu Se deve colocar o cálculo na TPU.