MLIR
21.0.0git
|
Go to the source code of this file.
Classes | |
struct | MlirLinalgContractionDimensions |
struct | MlirLinalgConvolutionDimensions |
Typedefs | |
typedef struct MlirLinalgContractionDimensions | MlirLinalgContractionDimensions |
typedef struct MlirLinalgConvolutionDimensions | MlirLinalgConvolutionDimensions |
Functions | |
MLIR_CAPI_EXPORTED 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) |
MLIR_DECLARE_CAPI_DIALECT_REGISTRATION (Linalg, linalg) | |
typedef struct MlirLinalgContractionDimensions MlirLinalgContractionDimensions |
typedef struct MlirLinalgConvolutionDimensions MlirLinalgConvolutionDimensions |
MLIR_DECLARE_CAPI_DIALECT_REGISTRATION | ( | Linalg | , |
linalg | |||
) |
MLIR_CAPI_EXPORTED void mlirLinalgFillBuiltinNamedOpRegion | ( | MlirOperation | mlirOp | ) |
Apply the special region builder for the builtin named Linalg op.
Assert that mlirOp
is a 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().
MLIR_CAPI_EXPORTED MlirAttribute mlirLinalgGetIndexingMapsAttribute | ( | MlirOperation | op | ) |
Definition at line 124 of file Linalg.cpp.
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().
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().
MLIR_CAPI_EXPORTED bool mlirLinalgIsAContractionOp | ( | MlirOperation | op | ) |
Definition at line 44 of file Linalg.cpp.
References mlir::linalg::isaContractionOpInterface(), and unwrap().
MLIR_CAPI_EXPORTED bool mlirLinalgIsAConvolutionOp | ( | MlirOperation | op | ) |
Definition at line 78 of file Linalg.cpp.
References mlir::linalg::isaConvolutionOpInterface(), and unwrap().