Prelinearize.Options

public static class Prelinearize.Options

Optional attributes for Prelinearize

Public Methods

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 Prelinearize.Options layout (List<Long> layout)

Parameters
layout A vector holding the requested layout in minor-to-major sequence. If a layout attribute is passed but its values are all -1 the layout will be computed by the infeed operation.

public Prelinearize.Options shape (Shape shape)

Parameters
shape The shape of the tensor.