TPUReplicateMetadata

Clase public final TPUReplicateMetadata

Metadatos que indican cómo se debe replicar el cálculo de la TPU.

Esta operación contiene los metadatos comunes a las operaciones de un subgrafo de cálculo `tpu.replicate ()`.

Clases anidadas

clase TPUReplicateMetadata.Options Los atributos opcionales para TPUReplicateMetadata

Constantes

Cuerda OP_NAME El nombre de esta operación, como lo conoce el motor central de TensorFlow

Métodos públicos

estáticas TPUReplicateMetadata.Options
allowSoftPlacement (Boolean allowSoftPlacement)
estáticas TPUReplicateMetadata.Options
computationShape (List <Larga> computationShape)
estática TPUReplicateMetadata
crear ( Alcance alcance, largas numReplicas, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación TPUReplicateMetadata.
estáticas TPUReplicateMetadata.Options
deviceAssignment (List <Larga> deviceAssignment)
estáticas TPUReplicateMetadata.Options
hostComputeCore (List <String> hostComputeCore)
estáticas TPUReplicateMetadata.Options
numCoresPerReplica (Long numCoresPerReplica)
estáticas TPUReplicateMetadata.Options
paddingMap (List <String> paddingMap)
estáticas TPUReplicateMetadata.Options
stepMarkerLocation (String stepMarkerLocation)
estáticas TPUReplicateMetadata.Options
topología (topología String)
estáticas TPUReplicateMetadata.Options
useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)
estáticas TPUReplicateMetadata.Options
useTpu (Boolean useTpu)

Métodos heredados

Constantes

OP_NAME pública final static String

El nombre de esta operación, como lo conoce el motor central de TensorFlow

Valor constante: "TPUReplicateMetadata"

Métodos públicos

public static TPUReplicateMetadata.Options allowSoftPlacement (booleano allowSoftPlacement)

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

Parámetros
computationShape OBSOLETO. Utilice num_cores_per_replica en su lugar.

public static TPUReplicateMetadata crear ( Alcance alcance, Long numReplicas, Opciones ... Opciones)

Método de fábrica para crear una clase que envuelva una nueva operación TPUReplicateMetadata.

Parámetros
alcance alcance actual
numReplicas Número de réplicas del cálculo
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de TPUReplicateMetadata

public static TPUReplicateMetadata.Options deviceAssignment (List <Larga> deviceAssignment)

Parámetros
deviceAssignment La asignación de dispositivos para el cálculo.

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. Se utiliza para el paralelismo de modelos.

estáticas pública TPUReplicateMetadata.Options paddingMap (List <String> paddingMap)

public static TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

estáticas pública TPUReplicateMetadata.Options topología (topología de cadena)

Parámetros
topología TopologyProto que indica la topología del segmento del pod de TPU.

public static TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

public static TPUReplicateMetadata.Options useTpu (booleano useTpu)

Parámetros
useTpu Ya sea para colocar el cálculo en la TPU.