TPUPartitionedOutput.Options

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.