ReplicateMetadata

classe final pública ReplicateMetadata

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

Constantes

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

Métodos Públicos

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

public static ReplicateMetadata.Options permitirSoftPlacement (booleano permitirSoftPlacement)

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

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

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

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

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 ReplicateMetadata

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

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

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

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

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)

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

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

public static ReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

public static ReplicateMetadata.Options useTpu (booleano useTpu)

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