ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

TPUPartitionedOutput.Options

public static class TPUPartitionedOutput.Options

Optional attributes for TPUPartitionedOutput

Public Methods

TPUPartitionedOutput.Options
partitionDim(Long partitionDim)

Inherited Methods

Public Methods

public TPUPartitionedOutput.Options partitionDim (Long partitionDim)

Parameters
partitionDim An integer describles which dimension is partitioned.