XlaSplitND.Options

public static class XlaSplitND.Options

Optional attributes for XlaSplitND

Public Methods

XlaSplitND.Options

paddings(List<Long> paddings)

Inherited Methods

From class java.lang.Object

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.