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

Da classe org.tensorflow.op.PrimitiveOp

booleano final	igual a (objeto obj)
final int	código de hash ()
Operação	op () Retorna a Operation subjacente
String final	toString ()

Da classe java.lang.Object

boleano	igual a (objeto arg0)
Aula final<?>	getClass ()
int	código de hash ()
vazio final	notifique ()
vazio final	notifiqueTodos ()
Corda	toString ()
vazio final	espere (long arg0, int arg1)
vazio final	espere (arg0 longo)
vazio final	espere ()

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.