DialectLinalg.cpp File Reference

#include "mlir-c/Dialect/Linalg.h"
#include "mlir-c/IR.h"
#include "mlir/Bindings/Python/Nanobind.h"
#include "mlir/Bindings/Python/NanobindAdaptors.h"

Go to the source code of this file.

Functions
static std::optional< MlirLinalgContractionDimensions >	InferContractionDimensions (MlirOperation op)
static std::optional< MlirLinalgConvolutionDimensions >	InferConvolutionDimensions (MlirOperation op)
static void	populateDialectLinalgSubmodule (nb::module_ m)
	NB_MODULE (_mlirDialectsLinalg, m)

Function Documentation

InferContractionDimensions()

static std::optional<MlirLinalgContractionDimensions> InferContractionDimensions ( MlirOperation  op )

static

Definition at line 18 of file DialectLinalg.cpp.

References MlirLinalgContractionDimensions::batch, MlirLinalgContractionDimensions::k, MlirLinalgContractionDimensions::m, mlirAttributeIsNull(), mlirLinalgInferContractionDimensions(), and MlirLinalgContractionDimensions::n.

InferConvolutionDimensions()

static std::optional<MlirLinalgConvolutionDimensions> InferConvolutionDimensions ( MlirOperation  op )

static

Definition at line 32 of file DialectLinalg.cpp.

References MlirLinalgConvolutionDimensions::batch, MlirLinalgConvolutionDimensions::depth, MlirLinalgConvolutionDimensions::dilations, MlirLinalgConvolutionDimensions::filterLoop, MlirLinalgConvolutionDimensions::inputChannel, mlirAttributeIsNull(), mlirLinalgInferConvolutionDimensions(), MlirLinalgConvolutionDimensions::outputChannel, MlirLinalgConvolutionDimensions::outputImage, and MlirLinalgConvolutionDimensions::strides.

NB_MODULE()

NB_MODULE	(	_mlirDialectsLinalg	,
		m
	)

Definition at line 135 of file DialectLinalg.cpp.

populateDialectLinalgSubmodule()

static void populateDialectLinalgSubmodule ( nb::module_  m )

static

Definition at line 51 of file DialectLinalg.cpp.