mlir::linalg::ContractionDimensions Struct Reference

Positions of a Linalg op loops that correspond to different kinds of a contraction dimension. More...

#include "mlir/Dialect/Linalg/IR/LinalgInterfaces.h"

Public Attributes
SmallVector< unsigned, 2 >	batch
SmallVector< unsigned, 2 >	m
SmallVector< unsigned, 2 >	n
SmallVector< unsigned, 2 >	k

Detailed Description

Positions of a Linalg op loops that correspond to different kinds of a contraction dimension.

Definition at line 44 of file LinalgInterfaces.h.

Member Data Documentation

batch

SmallVector<unsigned, 2> mlir::linalg::ContractionDimensions::batch

Definition at line 45 of file LinalgInterfaces.h.

Referenced by inferContractionDimsImpl(), and mlirLinalgInferContractionDimensions().

k

SmallVector<unsigned, 2> mlir::linalg::ContractionDimensions::k

Definition at line 48 of file LinalgInterfaces.h.

Referenced by inferContractionDimsImpl(), and mlirLinalgInferContractionDimensions().

m

SmallVector<unsigned, 2> mlir::linalg::ContractionDimensions::m

Definition at line 46 of file LinalgInterfaces.h.

Referenced by inferContractionDimsImpl(), and mlirLinalgInferContractionDimensions().

n

SmallVector<unsigned, 2> mlir::linalg::ContractionDimensions::n

Definition at line 47 of file LinalgInterfaces.h.

Referenced by inferContractionDimsImpl(), and mlirLinalgInferContractionDimensions().

---

The documentation for this struct was generated from the following file:

• include/mlir/Dialect/Linalg/IR/LinalgInterfaces.h