Linalg.cpp File Reference

#include "mlir-c/Dialect/Linalg.h"
#include "mlir/CAPI/AffineMap.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.
MLIR_CAPI_EXPORTED bool	mlirLinalgIsAContractionOp (MlirOperation op)
static MlirLinalgContractionDimensions	toContractionDimensions (MLIRContext *ctx, const linalg::ContractionDimensions &dims)
MLIR_CAPI_EXPORTED MlirLinalgContractionDimensions	mlirLinalgInferContractionDimensions (MlirOperation op)
MLIR_CAPI_EXPORTED MlirLinalgContractionDimensions	mlirLinalgInferContractionDimensionsFromMaps (const MlirAffineMap *indexingMaps, size_t numMaps)
MLIR_CAPI_EXPORTED bool	mlirLinalgIsAConvolutionOp (MlirOperation op)
static MlirLinalgConvolutionDimensions	toConvolutionDimensions (MLIRContext *ctx, const linalg::ConvolutionDimensions &dims)
MLIR_CAPI_EXPORTED MlirLinalgConvolutionDimensions	mlirLinalgInferConvolutionDimensions (MlirOperation op)
MLIR_CAPI_EXPORTED MlirLinalgConvolutionDimensions	mlirLinalgInferConvolutionDimensionsFromMaps (const MlirAffineMap *indexingMaps, size_t numMaps)
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 19 of file Linalg.cpp.

References b, 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(), and unwrap().

Referenced by mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::linalg::populateDialectLinalgSubmodule().

mlirLinalgGetIndexingMapsAttribute()

MLIR_CAPI_EXPORTED MlirAttribute mlirLinalgGetIndexingMapsAttribute

(

MlirOperation

op

)

Definition at line 155 of file Linalg.cpp.

References ArrayAttr(), unwrap(), and wrap().

Referenced by mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::linalg::populateDialectLinalgSubmodule().

mlirLinalgInferContractionDimensions()

MLIR_CAPI_EXPORTED MlirLinalgContractionDimensions mlirLinalgInferContractionDimensions

(

MlirOperation

op

)

Definition at line 61 of file Linalg.cpp.

References mlir::linalg::inferContractionDims(), toContractionDimensions(), and unwrap().

Referenced by mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::linalg::InferContractionDimensions().

mlirLinalgInferContractionDimensionsFromMaps()

MLIR_CAPI_EXPORTED MlirLinalgContractionDimensions mlirLinalgInferContractionDimensionsFromMaps	(	const MlirAffineMap *	indexingMaps,
		size_t	numMaps )

Definition at line 77 of file Linalg.cpp.

References mlir::linalg::inferContractionDims(), toContractionDimensions(), and unwrap().

Referenced by mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::linalg::populateDialectLinalgSubmodule().

mlirLinalgInferConvolutionDimensions()

MLIR_CAPI_EXPORTED MlirLinalgConvolutionDimensions mlirLinalgInferConvolutionDimensions

(

MlirOperation

op

)

Definition at line 121 of file Linalg.cpp.

References mlir::linalg::inferConvolutionDims(), toConvolutionDimensions(), and unwrap().

Referenced by mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::linalg::InferConvolutionDimensions().

mlirLinalgInferConvolutionDimensionsFromMaps()

MLIR_CAPI_EXPORTED MlirLinalgConvolutionDimensions mlirLinalgInferConvolutionDimensionsFromMaps	(	const MlirAffineMap *	indexingMaps,
		size_t	numMaps )

Definition at line 135 of file Linalg.cpp.

References getContext(), mlir::linalg::inferConvolutionDims(), toConvolutionDimensions(), and unwrap().

Referenced by mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::linalg::populateDialectLinalgSubmodule().

mlirLinalgIsAContractionOp()

MLIR_CAPI_EXPORTED bool mlirLinalgIsAContractionOp

(

MlirOperation

op

)

Definition at line 45 of file Linalg.cpp.

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

Referenced by mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::linalg::populateDialectLinalgSubmodule().

mlirLinalgIsAConvolutionOp()

MLIR_CAPI_EXPORTED bool mlirLinalgIsAConvolutionOp

(

MlirOperation

op

)

Definition at line 97 of file Linalg.cpp.

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

Referenced by mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::linalg::populateDialectLinalgSubmodule().

toContractionDimensions()

MlirLinalgContractionDimensions toContractionDimensions ( MLIRContext * ctx, const linalg::ContractionDimensions & dims )

static

Definition at line 52 of file Linalg.cpp.

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

Referenced by mlirLinalgInferContractionDimensions(), and mlirLinalgInferContractionDimensionsFromMaps().

toConvolutionDimensions()

MlirLinalgConvolutionDimensions toConvolutionDimensions ( MLIRContext * ctx, const linalg::ConvolutionDimensions & dims )

static

Definition at line 106 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< int64_t >::get(), mlir::linalg::ConvolutionDimensions::inputChannel, mlir::linalg::ConvolutionDimensions::outputChannel, mlir::linalg::ConvolutionDimensions::outputImage, mlir::linalg::ConvolutionDimensions::strides, and wrap().

Referenced by mlirLinalgInferConvolutionDimensions(), and mlirLinalgInferConvolutionDimensionsFromMaps().