TransformUtils.h File Reference

#include "mlir/Dialect/Tensor/IR/Tensor.h"
#include "mlir/IR/PatternMatch.h"

Go to the source code of this file.

Classes
class	mlir::tensor::ExtractSliceFromCollapseHelper
	This class assists with generating IR required to materialize an arbitrary-sized slice from the result of a CollapseShapeOp. More...

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

Functions
FailureOr< Operation * >	mlir::tensor::simplifyCollapseShapeWithRankReducingExtractSlice (tensor::CollapseShapeOp op, RewriterBase &rewriter)
	Tries to simplify a tensor.collapse_shape operation by inserting a single rank-reducing tensor.extract_slice operation.