Aviso: esta API está obsoleta e será removida em uma versão futura do TensorFlow depois que a substituição estiver estável.

TPUReplicateMetadata

public final class TPUReplicateMetadata

Metadados que indicam como a computação da TPU deve ser replicada.

Esta operação mantém os metadados comuns às operações de um subgráfico de computação `tpu.replicate ()`.

Classes aninhadas

classe TPUReplicateMetadata.Options Atributos opcionais para TPUReplicateMetadata

Métodos Públicos

estáticos TPUReplicateMetadata.Options
allowSoftPlacement (booleano allowSoftPlacement)
estáticos TPUReplicateMetadata.Options
computationShape (List <longo> computationShape)
estática TPUReplicateMetadata
criar ( Scope escopo, numReplicas longas, Options ... Opções)
Método de fábrica para criar uma classe envolvendo uma nova operação TPUReplicateMetadata.
estáticos TPUReplicateMetadata.Options
deviceAssignment (List <longo> deviceAssignment)
estáticos TPUReplicateMetadata.Options
hostComputeCore (List <String> hostComputeCore)
estáticos TPUReplicateMetadata.Options
numCoresPerReplica (Long numCoresPerReplica)
estáticos TPUReplicateMetadata.Options
paddingMap (List <String> paddingMap)
estáticos TPUReplicateMetadata.Options
stepMarkerLocation (String stepMarkerLocation)
estáticos TPUReplicateMetadata.Options
topologia (topologia String)
estáticos TPUReplicateMetadata.Options
useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)
estáticos TPUReplicateMetadata.Options
useTpu (booleano useTpu)

Métodos herdados

Métodos Públicos

public static TPUReplicateMetadata.Options allowSoftPlacement (Boolean allowSoftPlacement)

public static TPUReplicateMetadata.Options computationShape (List <longo> computationShape)

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

public static TPUReplicateMetadata criar ( Scope escopo, Long numReplicas, Options ... Opções)

Método de fábrica para criar uma classe envolvendo uma nova operação TPUReplicateMetadata.

Parâmetros
alcance escopo atual
numReplicas Número de réplicas do cálculo
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de TPUReplicateMetadata

public static TPUReplicateMetadata.Options deviceAssignment (List <longo> deviceAssignment)

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

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

public static TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

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

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

public static TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

public static TPUReplicateMetadata.Options topologia (topologia String)

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

public static TPUReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

public static TPUReplicateMetadata.Options useTpu (Boolean useTpu)

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