TPUReplicateMetadata.Options

TPUReplicateMetadata.Options estáticos de clase pública

Los atributos opcionales para TPUReplicateMetadata

Métodos públicos

TPUReplicateMetadata.Options
allowSoftPlacement (Boolean allowSoftPlacement)
TPUReplicateMetadata.Options
computationShape (List <Larga> computationShape)
TPUReplicateMetadata.Options
deviceAssignment (List <Larga> deviceAssignment)
TPUReplicateMetadata.Options
hostComputeCore (List <String> hostComputeCore)
TPUReplicateMetadata.Options
numCoresPerReplica (Long numCoresPerReplica)
TPUReplicateMetadata.Options
paddingMap (List <String> paddingMap)
TPUReplicateMetadata.Options
stepMarkerLocation (String stepMarkerLocation)
TPUReplicateMetadata.Options
topología (topología String)
TPUReplicateMetadata.Options
useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)
TPUReplicateMetadata.Options
useTpu (Boolean useTpu)

Métodos heredados

Métodos públicos

pública TPUReplicateMetadata.Options allowSoftPlacement (booleano allowSoftPlacement)

pública TPUReplicateMetadata.Options computationShape (List <Larga> computationShape)

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

pública TPUReplicateMetadata.Options deviceAssignment (List <Larga> deviceAssignment)

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

pública TPUReplicateMetadata.Options hostComputeCore (List <String> hostComputeCore)

pública TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

Parámetros
numCoresPerReplica Número de núcleos por réplica. Se utiliza para el paralelismo de modelos.

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

pública TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

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.

pública TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

pública TPUReplicateMetadata.Options useTpu (booleano useTpu)

Parámetros
useTpu Ya sea para colocar el cálculo en la TPU.
,
TPUReplicateMetadata.Options estáticos de clase pública

Los atributos opcionales para TPUReplicateMetadata

Métodos públicos

TPUReplicateMetadata.Options
allowSoftPlacement (Boolean allowSoftPlacement)
TPUReplicateMetadata.Options
computationShape (List <Larga> computationShape)
TPUReplicateMetadata.Options
deviceAssignment (List <Larga> deviceAssignment)
TPUReplicateMetadata.Options
hostComputeCore (List <String> hostComputeCore)
TPUReplicateMetadata.Options
numCoresPerReplica (Long numCoresPerReplica)
TPUReplicateMetadata.Options
paddingMap (List <String> paddingMap)
TPUReplicateMetadata.Options
stepMarkerLocation (String stepMarkerLocation)
TPUReplicateMetadata.Options
topología (topología String)
TPUReplicateMetadata.Options
useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)
TPUReplicateMetadata.Options
useTpu (Boolean useTpu)

Métodos heredados

Métodos públicos

pública TPUReplicateMetadata.Options allowSoftPlacement (booleano allowSoftPlacement)

pública TPUReplicateMetadata.Options computationShape (List <Larga> computationShape)

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

pública TPUReplicateMetadata.Options deviceAssignment (List <Larga> deviceAssignment)

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

pública TPUReplicateMetadata.Options hostComputeCore (List <String> hostComputeCore)

pública TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

Parámetros
numCoresPerReplica Número de núcleos por réplica. Se utiliza para el paralelismo de modelos.

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

pública TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

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.

pública TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

pública TPUReplicateMetadata.Options useTpu (booleano useTpu)

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