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