|
bool | mlir::linalg::detail::canOpOperandsBeDroppedImpl (linalg::LinalgOp linalgOp, ArrayRef< OpOperand * > droppedOperands) |
| Implementation of the method that check if given operands can be dropped, i.e. More...
|
|
FailureOr< ContractionDimensions > | mlir::linalg::inferContractionDims (LinalgOp linalgOp) |
| Find at least 2 parallel (m and n) and 1 reduction (k) dimension candidates that form a matmul subcomputation within linalgOp . More...
|
|
FailureOr< ContractionDimensions > | mlir::linalg::inferContractionDims (ArrayRef< AffineMap > indexingMaps) |
|
bool | mlir::linalg::isaContractionOpInterface (LinalgOp linalgOp) |
| Checks whether linalgOp conforms to ContractionOpInterface. More...
|
|
FailureOr< ConvolutionDimensions > | mlir::linalg::inferConvolutionDims (LinalgOp linalgOp) |
| Find at least 1 parallel (output_image) and reduction (filter_loop) dimension candidates that form a convolution subcomputation within linalgOp . More...
|
|
bool | mlir::linalg::isaConvolutionOpInterface (LinalgOp linalgOp, bool allowEmptyConvolvedDims=false) |
| Checks whether linalgOp conforms to ConvolutionOpInterface. More...
|
|
bool | mlir::linalg::isaCopyOpInterface (LinalgOp linalgOp) |
| Checks whether linalgOp is semantically equivalent to a linalg.copyOp . More...
|
|
std::optional< SmallVector< int64_t > > | mlir::linalg::isaBroadcastOpInterface (GenericOp genericOp) |
| Checks whether genericOp is semantically equivalent to a linalg.broadcast . More...
|
|
std::optional< SmallVector< int64_t > > | mlir::linalg::isaTransposeOpInterface (GenericOp genericOp) |
| Checks whether genericOp is semantically equivalent to a linalg.transpose . More...
|
|
bool | mlir::linalg::isaElemwiseSingleUnaryOpInterface (GenericOp genericOp) |
| Checks whether a given genericOp is semantically equivalent to a single linalgelementwise unary op. More...
|
|
bool | mlir::linalg::isaElemwiseSingleBinaryOpInterface (GenericOp genericOp) |
| Checks whether genericOp is semantically equivalent to a single linalg elementwise binary op e.g. More...
|
|
std::optional< Value > | mlir::linalg::isaFillOpInterface (GenericOp genericOp) |
| Checks whether genericOp is semantically equivalent to a linalg.fill . More...
|
|
bool | mlir::linalg::detail::isContractionBody (Block &block, function_ref< bool(Operation *, Operation *)> isaPair, llvm::raw_ostream &errs=mlir::thread_safe_nulls()) |
| Returns true if the block contains a contraction of the following form: More...
|
|
MatchContractionResult | mlir::linalg::detail::isContractionInterfaceImpl (Operation *op, ContractionDimensions *dimensions=nullptr) |
| Checks whether op conforms to ContractionOpInterface and populates dimensions with indexes of the different kinds of dimensions when present. More...
|
|
StringRef | mlir::linalg::detail::getMatchContractionMessage (MatchContractionResult res) |
| Returns the error message corresponding to the contraction checking return code. More...
|
|
MatchConvolutionResult | mlir::linalg::detail::isConvolutionInterfaceImpl (Operation *op, ConvolutionDimensions *dimensions=nullptr, bool allowEmptyConvolvedDims=false) |
| Checks whether op conforms to ConvolutionOpInterface and populates dimensions with indexes of the different kinds of dimensions when present. More...
|
|
StringRef | mlir::linalg::detail::getMatchConvolutionMessage (MatchConvolutionResult res) |
| Returns the error message corresponding to the convolution checking return code. More...
|
|
LogicalResult | mlir::linalg::detail::verifyContractionInterface (Operation *op) |
| Verify that op conforms to ContractionOpInterface. More...
|
|
LogicalResult | mlir::linalg::detail::verifyConvolutionInterface (Operation *op) |
| Verify that op conforms to the ConvolutionOpInterface. More...
|
|
LogicalResult | mlir::linalg::detail::verifyFillInterface (Operation *op) |
| Verify that op conforms to the FillOpInterface. More...
|
|
LogicalResult | mlir::linalg::detail::verifyStructuredOpInterface (Operation *op) |
| Verify that op conforms to the invariants of StructuredOpInterface. More...
|
|