public static class Prelinearize.Options extends Object
Prelinearize
Modifier and Type | Method and Description |
---|---|
Prelinearize.Options |
layout(List<Long> layout) |
Prelinearize.Options |
shape(Shape shape) |
public Prelinearize.Options shape(Shape shape)
shape
- The shape of the tensor.public Prelinearize.Options layout(List<Long> layout)
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.Copyright © 2022. All rights reserved.