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 | |
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 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 numCoresPerReplica (Long numCoresPerReplica)
Parâmetros
numCoresPerReplica | Número de núcleos por réplica. Usado para paralelismo de modelo. |
---|
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 useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)
public static TPUReplicateMetadata.Options useTpu (Boolean useTpu)
Parâmetros
use Tpu | Se o cálculo deve ser colocado na TPU. |
---|