org.tensorflow.op.core

Class TPUReplicateMetadata.Options

java.lang.Object

org.tensorflow.op.core.TPUReplicateMetadata.Options

Enclosing class:

TPUReplicateMetadata

public static class TPUReplicateMetadata.Options
extends Object

Optional attributes for TPUReplicateMetadata

Method Summary

Modifier and Type	Method and Description
TPUReplicateMetadata.Options	allowSoftPlacement(Boolean allowSoftPlacement)
TPUReplicateMetadata.Options	computationShape(List<Long> computationShape)
TPUReplicateMetadata.Options	deviceAssignment(List<Long> 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	topology(String topology)
TPUReplicateMetadata.Options	useTpu(Boolean useTpu)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

numCoresPerReplica

public TPUReplicateMetadata.Options numCoresPerReplica(Long numCoresPerReplica)

Parameters:

numCoresPerReplica - Number of cores per replica. Used for model parallelism.

topology

public TPUReplicateMetadata.Options topology(String topology)

Parameters:

topology - TopologyProto indicating the topology of the TPU pod slice.

useTpu

public TPUReplicateMetadata.Options useTpu(Boolean useTpu)

Parameters:

useTpu - Whether to place the computation on the TPU.

deviceAssignment

public TPUReplicateMetadata.Options deviceAssignment(List<Long> deviceAssignment)

Parameters:

deviceAssignment - The assignment of devices for the computation.

computationShape

public TPUReplicateMetadata.Options computationShape(List<Long> computationShape)

Parameters:

computationShape - DEPRECATED. Use num_cores_per_replica instead.

hostComputeCore

public TPUReplicateMetadata.Options hostComputeCore(List<String> hostComputeCore)

Parameters:

hostComputeCore -

paddingMap

public TPUReplicateMetadata.Options paddingMap(List<String> paddingMap)

Parameters:

paddingMap -

stepMarkerLocation

public TPUReplicateMetadata.Options stepMarkerLocation(String stepMarkerLocation)

Parameters:

stepMarkerLocation -

allowSoftPlacement

public TPUReplicateMetadata.Options allowSoftPlacement(Boolean allowSoftPlacement)

Parameters:

allowSoftPlacement -