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