ReplicateMetadata

ReplicateMetadata classe final pública

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 ReplicateMetadata.Options Atributos opcionais para ReplicateMetadata

Constantes

Fragmento OP_NAME O nome desta operação, conforme conhecido pelo motor principal TensorFlow

Métodos Públicos

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

Métodos herdados

Constantes

nome_op final String public static

O nome desta operação, conforme conhecido pelo motor principal TensorFlow

Valor constante: "TPUReplicateMetadata"

Métodos Públicos

public static ReplicateMetadata.Options allowSoftPlacement (Boolean allowSoftPlacement)

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

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

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

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

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 ReplicateMetadata

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

Parâmetros
deviceAssignment A atribuição de dispositivos para o cálculo.

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

public static ReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

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

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

public static ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

public static ReplicateMetadata.Options topologia (topologia String)

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

public static ReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

public static ReplicateMetadata.Options useTpu (Boolean useTpu)

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