Uses of Class
org.tensorflow.op.xla.SpmdFullToShardShape.Options

Packages that use SpmdFullToShardShape.Options

Package	Description
org.tensorflow.op
org.tensorflow.op.xla

Uses of SpmdFullToShardShape.Options in org.tensorflow.op

Methods in org.tensorflow.op with parameters of type SpmdFullToShardShape.Options

Modifier and Type	Method and Description
<T extends TType> SpmdFullToShardShape<T>	XlaOps.spmdFullToShardShape(Operand<T> input, String manualSharding, SpmdFullToShardShape.Options... options) An op used by XLA SPMD partitioner to switch from automatic partitioning to manual partitioning.

Uses of SpmdFullToShardShape.Options in org.tensorflow.op.xla

Methods in org.tensorflow.op.xla that return SpmdFullToShardShape.Options

Modifier and Type	Method and Description
static SpmdFullToShardShape.Options	SpmdFullToShardShape.dim(Long dim) Sets the dim option.
SpmdFullToShardShape.Options	SpmdFullToShardShape.Options.dim(Long dim) Sets the dim option.
static SpmdFullToShardShape.Options	SpmdFullToShardShape.unspecifiedDims(List<Long> unspecifiedDims) Sets the unspecifiedDims option.
SpmdFullToShardShape.Options	SpmdFullToShardShape.Options.unspecifiedDims(List<Long> unspecifiedDims) Sets the unspecifiedDims option.
static SpmdFullToShardShape.Options	SpmdFullToShardShape.unspecifiedDims(Long... unspecifiedDims) Sets the unspecifiedDims option.
SpmdFullToShardShape.Options	SpmdFullToShardShape.Options.unspecifiedDims(Long... unspecifiedDims) Sets the unspecifiedDims option.

Methods in org.tensorflow.op.xla with parameters of type SpmdFullToShardShape.Options

Modifier and Type	Method and Description
static <T extends TType> SpmdFullToShardShape<T>	SpmdFullToShardShape.create(Scope scope, Operand<T> input, String manualSharding, SpmdFullToShardShape.Options... options) Factory method to create a class wrapping a new XlaSpmdFullToShardShape operation.