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

TPUPartitionedInput.Options

public static class TPUPartitionedInput.Options

Optional attributes for TPUPartitionedInput

Public Methods

TPUPartitionedInput.Options
partitionDim(Long partitionDim)

Inherited Methods

Public Methods

public TPUPartitionedInput.Options partitionDim (Long partitionDim)

Parameters
partitionDim An integer describles which dimension is partitioned. -1 means those inputs are replicated.