Aprenda o que há de mais recente em aprendizado de máquina, IA generativa e muito mais no WiML Symposium 2023
Registre-se
TPUReplicateMetadata.Options
Mantenha tudo organizado com as coleções
Salve e categorize o conteúdo com base nas suas preferências.
Métodos herdados
Da classe java.lang.Object boleano | equals (arg0 Object) |
aula final <?> | getClass () |
int | hashCode () |
vazio final | notificar () |
vazio final | notifyAll () |
Fragmento | toString () |
vazio final | wait (longa arg0, int arg1) |
vazio final | wait (arg0 de comprimento) |
vazio final | wait () |
Métodos Públicos
Parâmetros
computationShape | DESCONTINUADA. Em vez disso, use num_cores_per_replica. |
---|
Parâmetros
deviceAssignment | A atribuição de dispositivos para a computação. |
---|
Parâmetros
numCoresPerReplica | Número de núcleos por réplica. Usado para paralelismo de modelo. |
---|
Parâmetros
topologia | TopologyProto indicando a topologia da fatia do pod de TPU. |
---|
Parâmetros
useTpu | Se deve colocar o cálculo na TPU. |
---|
Exceto em caso de indicação contrária, o conteúdo desta página é licenciado de acordo com a Licença de atribuição 4.0 do Creative Commons, e as amostras de código são licenciadas de acordo com a Licença Apache 2.0. Para mais detalhes, consulte as políticas do site do Google Developers. Java é uma marca registrada da Oracle e/ou afiliadas.
Última atualização 2021-11-29 UTC.
[null,null,["Última atualização 2021-11-29 UTC."],[],[],null,["# TPUReplicateMetadata.Options\n\npublic static class **TPUReplicateMetadata.Options** \nOptional attributes for [TPUReplicateMetadata](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata) \n\n### Public Methods\n\n|-------------------------------------------------------------------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options) | [allowSoftPlacement](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options#allowSoftPlacement(java.lang.Boolean))(Boolean allowSoftPlacement) |\n| [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options) | [computationShape](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options#computationShape(java.util.List\u003cjava.lang.Long\u003e))(List\\\u003cLong\\\u003e computationShape) |\n| [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options) | [deviceAssignment](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options#deviceAssignment(java.util.List\u003cjava.lang.Long\u003e))(List\\\u003cLong\\\u003e deviceAssignment) |\n| [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options) | [hostComputeCore](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options#hostComputeCore(java.util.List\u003cjava.lang.String\u003e))(List\\\u003cString\\\u003e hostComputeCore) |\n| [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options) | [numCoresPerReplica](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options#numCoresPerReplica(java.lang.Long))(Long numCoresPerReplica) |\n| [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options) | [paddingMap](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options#paddingMap(java.util.List\u003cjava.lang.String\u003e))(List\\\u003cString\\\u003e paddingMap) |\n| [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options) | [stepMarkerLocation](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options#stepMarkerLocation(java.lang.String))(String stepMarkerLocation) |\n| [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options) | [topology](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options#topology(java.lang.String))(String topology) |\n| [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options) | [useSpmdForXlaPartitioning](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options#useSpmdForXlaPartitioning(java.lang.Boolean))(Boolean useSpmdForXlaPartitioning) |\n| [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options) | [useTpu](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options#useTpu(java.lang.Boolean))(Boolean useTpu) |\n\n### Inherited Methods\n\nFrom class java.lang.Object \n\n|------------------|---------------------------|\n| boolean | equals(Object arg0) |\n| final Class\\\u003c?\\\u003e | getClass() |\n| int | hashCode() |\n| final void | notify() |\n| final void | notifyAll() |\n| String | toString() |\n| final void | wait(long arg0, int arg1) |\n| final void | wait(long arg0) |\n| final void | wait() |\n\nPublic Methods\n--------------\n\n#### public [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options)\n**allowSoftPlacement**\n(Boolean allowSoftPlacement)\n\n\u003cbr /\u003e\n\n#### public [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options)\n**computationShape**\n(List\\\u003cLong\\\u003e computationShape)\n\n\u003cbr /\u003e\n\n##### Parameters\n\n| computationShape | DEPRECATED. Use num_cores_per_replica instead. |\n|------------------|------------------------------------------------|\n\n#### public [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options)\n**deviceAssignment**\n(List\\\u003cLong\\\u003e deviceAssignment)\n\n\u003cbr /\u003e\n\n##### Parameters\n\n| deviceAssignment | The assignment of devices for the computation. |\n|------------------|------------------------------------------------|\n\n#### public [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options)\n**hostComputeCore**\n(List\\\u003cString\\\u003e hostComputeCore)\n\n\u003cbr /\u003e\n\n#### public [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options)\n**numCoresPerReplica**\n(Long numCoresPerReplica)\n\n\u003cbr /\u003e\n\n##### Parameters\n\n| numCoresPerReplica | Number of cores per replica. Used for model parallelism. |\n|--------------------|----------------------------------------------------------|\n\n#### public [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options)\n**paddingMap**\n(List\\\u003cString\\\u003e paddingMap)\n\n\u003cbr /\u003e\n\n#### public [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options)\n**stepMarkerLocation**\n(String stepMarkerLocation)\n\n\u003cbr /\u003e\n\n#### public [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options)\n**topology**\n(String topology)\n\n\u003cbr /\u003e\n\n##### Parameters\n\n| topology | TopologyProto indicating the topology of the TPU pod slice. |\n|----------|-------------------------------------------------------------|\n\n#### public [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options)\n**useSpmdForXlaPartitioning**\n(Boolean useSpmdForXlaPartitioning)\n\n\u003cbr /\u003e\n\n#### public [TPUReplicateMetadata.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/TPUReplicateMetadata.Options)\n**useTpu**\n(Boolean useTpu)\n\n\u003cbr /\u003e\n\n##### Parameters\n\n| useTpu | Whether to place the computation on the TPU. |\n|--------|----------------------------------------------|"]]