Attend the Women in ML Symposium on December 7 Register now

XlaSplitND.Options

public static class XlaSplitND.Options

Optional attributes for XlaSplitND

Public Methods

XlaSplitND.Options
paddings(List<Long> paddings)

Inherited Methods

Public Methods

public XlaSplitND.Options paddings (List<Long> paddings)

Parameters
paddings Optional list of right paddings per dimension of input tensor to apply before splitting. This can be used to make a dimension evenly divisible.