MLIR
22.0.0git
|
Transformation to drop unit-extent dimensions from linalg.generic
operations.
More...
#include "mlir/Dialect/Linalg/Transforms/Transforms.h"
Public Types | |
enum class | RankReductionStrategy { ReassociativeReshape , ExtractInsertSlice } |
using | ControlFnTy = std::function< SmallVector< unsigned >(Operation *)> |
Public Attributes | |
RankReductionStrategy | rankReductionStrategy |
ControlFnTy | controlFn |
Transformation to drop unit-extent dimensions from linalg.generic
operations.
Definition at line 522 of file Transforms.h.
using mlir::linalg::ControlDropUnitDims::ControlFnTy = std::function<SmallVector<unsigned>(Operation *)> |
Definition at line 528 of file Transforms.h.
Enumerator | |
---|---|
ReassociativeReshape | |
ExtractInsertSlice |
Definition at line 523 of file Transforms.h.
ControlFnTy mlir::linalg::ControlDropUnitDims::controlFn |
Definition at line 529 of file Transforms.h.
RankReductionStrategy mlir::linalg::ControlDropUnitDims::rankReductionStrategy |
Definition at line 525 of file Transforms.h.