MLIR 22.0.0git
Utils.h File Reference

Go to the source code of this file.

Namespaces

namespace  mlir
 Include the generated interface declarations.
namespace  mlir::tensor

Functions

PadOp mlir::tensor::createPadHighOp (RankedTensorType resType, Value source, Value pad, bool nofold, Location loc, OpBuilder &builder, ValueRange dynOutDims={})
SmallVector< Valuemlir::tensor::createDynamicDimValues (OpBuilder &b, Location loc, Value rankedTensor)
FailureOr< RankedTensorType > mlir::tensor::computeTransposedType (RankedTensorType rankedTensorType, ArrayRef< int64_t > transposeVector)
 Returns the transposed rankedTensorType if transposeVector is non-empty.
CollapseShapeOp mlir::tensor::dropGivenUnitDims (OpBuilder &b, Location loc, Value src, const llvm::SmallBitVector &dropDims)
 Create tensor.collapse_shape to drop unit dimensions in dropDims in tensor src.
bool mlir::tensor::isCastLikeInsertSliceOp (InsertSliceOp op)
 A tensor.insert_slice is a cast-like operation if it merely rank-extends the source tensor or inserts the source tensor into a destination tensor with the same shape.
bool mlir::tensor::isCastLikeExtractSliceOp (ExtractSliceOp op)
 A tensor.extract_slice is a cast-like operation if it merely rank-reduces unit dimensions of the source tensor or extracts the entire source tensor.