Attend the Women in ML Symposium on December 7 Register now

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.