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 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 numCoresPerReplica (Long numCoresPerReplica)
Parâmetros
numCoresPerReplica | Número de núcleos por réplica. Usado para paralelismo de modelo. |
---|
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. |
---|