MLIR
20.0.0git
|
#include "mlir/Dialect/Linalg/Transforms/Transforms.h"
Public Types | |
enum class | CopyBackOp : int8_t { None = 0 , BufferizationMaterializeInDestination = 1 , LinalgCopy = 2 } |
Public Member Functions | |
LinalgPaddingOptions & | setPaddingValues (ArrayRef< Attribute > pv) |
LinalgPaddingOptions & | setPaddingDimensions (ArrayRef< int64_t > pd) |
LinalgPaddingOptions & | setPadToMultipleOf (ArrayRef< int64_t > m) |
LinalgPaddingOptions & | setPackPaddings (ArrayRef< bool > pp) |
LinalgPaddingOptions & | setHoistPaddings (ArrayRef< int64_t > hp) |
LinalgPaddingOptions & | setTransposePaddings (ArrayRef< SmallVector< int64_t >> tp) |
LinalgPaddingOptions & | setCopyBackOp (CopyBackOp op) |
Public Attributes | |
SmallVector< Attribute > | paddingValues |
A padding value for every operand. More... | |
SmallVector< int64_t > | paddingDimensions |
A list of iterator dimensions to pad. More... | |
std::optional< SmallVector< int64_t > > | padToMultipleOf |
A list of multiples to which each padding dimension should be padded to. More... | |
SmallVector< bool > | packPaddings |
A flag for every operand to mark the PadOp as nofold which enables packing for statically shaped operands. More... | |
SmallVector< int64_t > | hoistPaddings |
A number of loops to hoist the PadOp out for every operand. More... | |
SmallVector< SmallVector< int64_t > > | transposePaddings |
A permutation vector for every operand used to transpose the packed PadOp results. More... | |
CopyBackOp | copyBackOp = CopyBackOp::BufferizationMaterializeInDestination |
The op to be used for copying the padded result to the original destination tensor. More... | |
Definition at line 278 of file Transforms.h.
|
strong |
Enumerator | |
---|---|
None | |
BufferizationMaterializeInDestination | |
LinalgCopy |
Definition at line 318 of file Transforms.h.
|
inline |
Definition at line 326 of file Transforms.h.
|
inline |
Definition at line 306 of file Transforms.h.
References hoistPaddings.
|
inline |
Definition at line 300 of file Transforms.h.
References packPaddings.
|
inline |
Definition at line 287 of file Transforms.h.
References paddingDimensions.
|
inline |
Definition at line 281 of file Transforms.h.
References paddingValues.
|
inline |
Definition at line 293 of file Transforms.h.
|
inline |
Definition at line 314 of file Transforms.h.
References transposePaddings.
CopyBackOp mlir::linalg::LinalgPaddingOptions::copyBackOp = CopyBackOp::BufferizationMaterializeInDestination |
The op to be used for copying the padded result to the original destination tensor.
Definition at line 325 of file Transforms.h.
SmallVector<int64_t> mlir::linalg::LinalgPaddingOptions::hoistPaddings |
A number of loops to hoist the PadOp out for every operand.
Definition at line 305 of file Transforms.h.
Referenced by setHoistPaddings().
SmallVector<bool> mlir::linalg::LinalgPaddingOptions::packPaddings |
A flag for every operand to mark the PadOp as nofold which enables packing for statically shaped operands.
Definition at line 299 of file Transforms.h.
Referenced by setPackPaddings().
SmallVector<int64_t> mlir::linalg::LinalgPaddingOptions::paddingDimensions |
A list of iterator dimensions to pad.
Definition at line 286 of file Transforms.h.
Referenced by setPaddingDimensions().
SmallVector<Attribute> mlir::linalg::LinalgPaddingOptions::paddingValues |
A padding value for every operand.
Definition at line 280 of file Transforms.h.
Referenced by setPaddingValues().
std::optional<SmallVector<int64_t> > mlir::linalg::LinalgPaddingOptions::padToMultipleOf |
A list of multiples to which each padding dimension should be padded to.
Definition at line 292 of file Transforms.h.
SmallVector<SmallVector<int64_t> > mlir::linalg::LinalgPaddingOptions::transposePaddings |
A permutation vector for every operand used to transpose the packed PadOp results.
Definition at line 312 of file Transforms.h.
Referenced by setTransposePaddings().