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. |
---|