ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Prelinearize.Options

public static class Prelinearize.Options

Optional attributes for Prelinearize

Public Methods

Prelinearize.Options
layout(List<Long> layout)
Prelinearize.Options
shape(Shape shape)

Inherited Methods

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.