TPUReplicateMetadata

classe final pública TPUReplicateMetadata

Metadados que indicam como o cálculo da TPU deve ser replicado.

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

Classes aninhadas

aula TPUReplicateMetadata.Options Atributos opcionais para TPUReplicateMetadata

Constantes

Corda OP_NAME O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow

Métodos Públicos

TPUReplicateMetadata.Options estático
permitirSoftPlacement (booleano permitirSoftPlacement)
TPUReplicateMetadata.Options estático
computationShape (List<Long> computationShape)
TPUReplicateMetadata estático
create ( Escopo , Long numReplicas, Opções... opções)
Método de fábrica para criar uma classe que agrupa uma nova operação TPUReplicateMetadata.
TPUReplicateMetadata.Options estático
deviceAssignment (Lista<Long> deviceAssignment)
TPUReplicateMetadata.Options estático
hostComputeCore (List<String> hostComputeCore)
TPUReplicateMetadata.Options estático
numCoresPerReplica (numCoresPerReplica longo)
TPUReplicateMetadata.Options estático
paddingMap (List<String> paddingMap)
TPUReplicateMetadata.Options estático
stepMarkerLocation (String stepMarkerLocation)
TPUReplicateMetadata.Options estático
topologia (topologia de string)
TPUReplicateMetadata.Options estático
useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)
TPUReplicateMetadata.Options estático
useTpu (booleano useTpu)

Métodos herdados

Constantes

String final estática pública OP_NAME

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

Valor constante: "TPUReplicateMetadata"

Métodos Públicos

público estático TPUReplicateMetadata.Options permitirSoftPlacement (booleano permitirSoftPlacement)

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

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

public static TPUReplicateMetadata create ( Escopo escopo, Long numReplicas, Opções... opções)

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

Parâmetros
escopo 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<Long> deviceAssignment)

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

público estático TPUReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

público estático TPUReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica longo)

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

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

público estático TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

Topologia pública estática TPUReplicateMetadata.Options (topologia String)

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

public static TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

público estático TPUReplicateMetadata.Options useTpu (booleano useTpu)

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