XlaSplitND.Options

public static class XlaSplitND.Options

Optional attributes for XlaSplitND

Public Methods

XlaSplitND.Options
paddings(List<Long> paddings)

Inherited Methods

boolean
equals(Object arg0)
final Class<?>
getClass()
int
hashCode()
final void
notify()
final void
notifyAll()
String
toString()
final void
wait(long arg0, int arg1)
final void
wait(long arg0)
final void
wait()

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.