Linalg.cpp File Reference

#include "mlir-c/Dialect/Linalg.h"
#include "mlir/CAPI/Registration.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h"

Go to the source code of this file.

Functions
void	mlirLinalgFillBuiltinNamedOpRegion (MlirOperation mlirOp)
	Apply the special region builder for the builtin named Linalg op. More...
MLIR_CAPI_EXPORTED bool	mlirLinalgIsAContractionOp (MlirOperation op)
MLIR_CAPI_EXPORTED MlirLinalgContractionDimensions	mlirLinalgInferContractionDimensions (MlirOperation op)
MLIR_CAPI_EXPORTED bool	mlirLinalgIsAConvolutionOp (MlirOperation op)
MLIR_CAPI_EXPORTED MlirLinalgConvolutionDimensions	mlirLinalgInferConvolutionDimensions (MlirOperation op)
MLIR_CAPI_EXPORTED MlirAttribute	mlirLinalgGetIndexingMapsAttribute (MlirOperation op)

Function Documentation

mlirLinalgFillBuiltinNamedOpRegion()

void mlirLinalgFillBuiltinNamedOpRegion

(

MlirOperation

mlirOp

)

Apply the special region builder for the builtin named Linalg op.

Assert that op is a builtin named Linalg op.

Definition at line 18 of file Linalg.cpp.

References mlir::OpBuilder::createBlock(), mlir::Operation::getAttrs(), mlir::Region::getBlocks(), mlir::Operation::getContext(), mlir::getElementTypeOrSelf(), mlir::Operation::getLoc(), mlir::Operation::getName(), mlir::Operation::getNumRegions(), mlir::Operation::getRegion(), mlir::OperationName::getStringRef(), mlir::OpBuilder::setInsertionPointToStart(), and unwrap().

mlirLinalgGetIndexingMapsAttribute()

MLIR_CAPI_EXPORTED MlirAttribute mlirLinalgGetIndexingMapsAttribute

(

MlirOperation

op

)

Definition at line 124 of file Linalg.cpp.

References unwrap(), and wrap().

mlirLinalgInferContractionDimensions()

MLIR_CAPI_EXPORTED MlirLinalgContractionDimensions mlirLinalgInferContractionDimensions

(

MlirOperation

op

)

Definition at line 51 of file Linalg.cpp.

References mlir::linalg::ContractionDimensions::batch, mlir::detail::DenseArrayAttrImpl< int32_t >::get(), mlir::linalg::inferContractionDims(), mlir::linalg::ContractionDimensions::k, mlir::linalg::ContractionDimensions::m, mlir::linalg::ContractionDimensions::n, unwrap(), and wrap().

Referenced by InferContractionDimensions().

mlirLinalgInferConvolutionDimensions()

MLIR_CAPI_EXPORTED MlirLinalgConvolutionDimensions mlirLinalgInferConvolutionDimensions

(

MlirOperation

op

)

Definition at line 87 of file Linalg.cpp.

References mlir::linalg::ConvolutionDimensions::batch, mlir::linalg::ConvolutionDimensions::depth, mlir::linalg::ConvolutionDimensions::dilations, mlir::linalg::ConvolutionDimensions::filterLoop, mlir::detail::DenseArrayAttrImpl< int32_t >::get(), mlir::detail::DenseArrayAttrImpl< T >::get(), mlir::linalg::inferConvolutionDims(), mlir::linalg::ConvolutionDimensions::inputChannel, mlir::linalg::ConvolutionDimensions::outputChannel, mlir::linalg::ConvolutionDimensions::outputImage, mlir::linalg::ConvolutionDimensions::strides, unwrap(), and wrap().

Referenced by InferConvolutionDimensions().

mlirLinalgIsAContractionOp()

MLIR_CAPI_EXPORTED bool mlirLinalgIsAContractionOp

(

MlirOperation

op

)

Definition at line 44 of file Linalg.cpp.

References mlir::linalg::isaContractionOpInterface(), and unwrap().

mlirLinalgIsAConvolutionOp()

MLIR_CAPI_EXPORTED bool mlirLinalgIsAConvolutionOp

(

MlirOperation

op

)

Definition at line 78 of file Linalg.cpp.

References mlir::linalg::isaConvolutionOpInterface(), and unwrap().