TPUReplicateMetadata

classe final pública TPUReplicateMetadata

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

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

Classes aninhadas

classe TPUReplicateMetadata.Options Atributos opcionais para TPUReplicateMetadata

Métodos públicos

static TPUReplicateMetadata.Options
allowSoftPlacement (booleano allowSoftPlacement)
static TPUReplicateMetadata.Options
forma de computação (List<Long> forma de computação)
static TPUReplicateMetadata
create ( Escopo do escopo, Long numReplicas, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação TPUReplicateMetadata.
static TPUReplicateMetadata.Options
deviceAssignment (List<Long> deviceAssignment)
static TPUReplicateMetadata.Options
hostComputeCore (List<> hostComputeCore)
static TPUReplicateMetadata.Options
numCoresPerReplica (long numCoresPerReplica)
static TPUReplicateMetadata.Options
paddingMap (List<String> paddingMap)
static TPUReplicateMetadata.Options
stepMarkerLocation (String stepMarkerLocation)
static TPUReplicateMetadata.Options
topologia (topologia de string)
static TPUReplicateMetadata.Options
tpuCompileOptionsProto (String tpuCompileOptionsProto)
static TPUReplicateMetadata.Options
useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)
static TPUReplicateMetadata.Options
useTpu (booleano useTpu)

Métodos Herdados

Métodos públicos

public static TPUReplicateMetadata.Options allowSoftPlacement (booleano allowSoftPlacement)

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

Parâmetros
computaçãoForma DESCONTINUADA. Em vez disso, use num_cores_per_replica.

public static TPUReplicateMetadata create ( Escopo do 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
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<Long> deviceAssignment)

Parâmetros
Atribuição de dispositivo 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<> paddingMap)

public static TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

topologia public static TPUReplicateMetadata.Options ( topologia de string)

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

public static TPUReplicateMetadata.Options tpuCompileOptionsProto (String tpuCompileOptionsProto)

public static TPUReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

public static TPUReplicateMetadata.Options useTpu (Boolean useTpu)

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