TPUPartitionedInput.Options

public static class TPUPartitionedInput.Options

Optional attributes for TPUPartitionedInput

Public Methods

TPUPartitionedInput.Options

partitionDim(Long partitionDim)

Inherited Methods

From class java.lang.Object

boolean	equals(Object arg0)
final Class<?>	getClass()
int	hashCode()
final void	notify()
final void	notifyAll()
String	toString()
final void	wait(long arg0, int arg1)
final void	wait(long arg0)
final void	wait()

Public Methods

public TPUPartitionedInput.Options partitionDim (Long partitionDim)

Parameters

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