MLIR
17.0.0git
|
This is a builder type that keeps local references to arguments. More...
#include "mlir/IR/BuiltinTypes.h"
Public Member Functions | |
Builder (RankedTensorType other) | |
Build from another RankedTensorType. More... | |
Builder (ArrayRef< int64_t > shape, Type elementType, Attribute encoding) | |
Build from scratch. More... | |
Builder & | setShape (ArrayRef< int64_t > newShape) |
Builder & | setElementType (Type newElementType) |
Builder & | setEncoding (Attribute newEncoding) |
Builder & | dropDim (unsigned pos) |
Erase a dim from shape @pos. More... | |
Builder & | insertDim (int64_t val, unsigned pos) |
Insert a val into shape @pos. More... | |
operator RankedTensorType () | |
This is a builder type that keeps local references to arguments.
Arguments that are passed into the builder must outlive the builder.
Definition at line 214 of file BuiltinTypes.h.
|
inlineexplicit |
Build from another RankedTensorType.
Definition at line 217 of file BuiltinTypes.h.
|
inline |
Build from scratch.
Definition at line 222 of file BuiltinTypes.h.
|
inline |
Erase a dim from shape @pos.
Definition at line 241 of file BuiltinTypes.h.
Referenced by mlir::linalg::DownscaleSizeOneWindowed2DConvolution< Conv2DOp, Conv1DOp >::returningMatchAndRewrite(), and mlir::linalg::DownscaleDepthwiseConv2DNhwcHwcOp::returningMatchAndRewrite().
|
inline |
Insert a val into shape @pos.
Definition at line 251 of file BuiltinTypes.h.
Referenced by mlir::linalg::splitReductionByScaling().
|
inline |
Definition at line 260 of file BuiltinTypes.h.
Definition at line 230 of file BuiltinTypes.h.
Definition at line 235 of file BuiltinTypes.h.
Definition at line 225 of file BuiltinTypes.h.
Referenced by computeTransposedType(), and permuteShape().