mlir::linalg::BatchMatmulTransposeAOp Class Reference

Specialization of linalg.batch_matmul op that has a transpose map on A. More...

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

Inheritance diagram for mlir::linalg::BatchMatmulTransposeAOp:

Static Public Member Functions
::mlir::TypeID	resolveTypeID ()
static void	build (OpBuilder &builder, OperationState &result, ValueRange inputs, ValueRange outputs, ArrayRef< NamedAttribute > attributes={})
	Build a transpose A matmul.
static BatchMatmulTransposeAOp	create (OpBuilder &builder, Location location, ValueRange inputs, ValueRange outputs, ArrayRef< NamedAttribute > attributes={})
static void	build (OpBuilder &builder, OperationState &result, TypeRange resultTensorTypes, ValueRange inputs, ValueRange outputs, ArrayRef< NamedAttribute > attributes={})
	Build a transpose A matmul with a specific result type.
static BatchMatmulTransposeAOp	create (OpBuilder &builder, Location location, TypeRange resultTensorTypes, ValueRange inputs, ValueRange outputs, ArrayRef< NamedAttribute > attributes={})
static void	build (OpBuilder &builder, OperationState &result, TypeRange resultTensorTypes, ValueRange inputs, ValueRange outputs, Attribute cast, ArrayRef< NamedAttribute > attributes={})
	Build a transpose A matmul with a specific result type and a cast type.
static BatchMatmulTransposeAOp	create (OpBuilder &builder, Location location, TypeRange resultTensorTypes, ValueRange inputs, ValueRange outputs, Attribute cast, ArrayRef< NamedAttribute > attributes={})
static bool	isDefaultIndexingMaps (Attribute attr)
	Checks if the affine map is the expected one for this operation.
static bool	classof (Operation *op)

Detailed Description

Specialization of linalg.batch_matmul op that has a transpose map on A.

Definition at line 251 of file Linalg.h.

Member Function Documentation

build() [1/3]

void mlir::linalg::linalg::BatchMatmulTransposeAOp::build ( OpBuilder & builder, OperationState & result, TypeRange resultTensorTypes, ValueRange inputs, ValueRange outputs, ArrayRef< NamedAttribute > attributes = {} )

static

Build a transpose A matmul with a specific result type.

Definition at line 4202 of file LinalgOps.cpp.

References buildMatmulOp(), and result.

build() [2/3]

void mlir::linalg::linalg::BatchMatmulTransposeAOp::build ( OpBuilder & builder, OperationState & result, TypeRange resultTensorTypes, ValueRange inputs, ValueRange outputs, Attribute cast, ArrayRef< NamedAttribute > attributes = {} )

static

Build a transpose A matmul with a specific result type and a cast type.

Definition at line 4223 of file LinalgOps.cpp.

References buildMatmulOp(), and result.

build() [3/3]

void mlir::linalg::linalg::BatchMatmulTransposeAOp::build ( OpBuilder & builder, OperationState & result, ValueRange inputs, ValueRange outputs, ArrayRef< NamedAttribute > attributes = {} )

static

Build a transpose A matmul.

Definition at line 4183 of file LinalgOps.cpp.

References buildMatmulOp(), and result.

Referenced by create(), create(), and create().

classof()

bool mlir::linalg::BatchMatmulTransposeAOp::classof ( Operation * op )

static

Definition at line 4245 of file LinalgOps.cpp.

References mlir::Operation::getAttr(), and isDefaultIndexingMaps().

create() [1/3]

BatchMatmulTransposeAOp mlir::linalg::BatchMatmulTransposeAOp::create ( OpBuilder & builder, Location location, TypeRange resultTensorTypes, ValueRange inputs, ValueRange outputs, ArrayRef< NamedAttribute > attributes = {} )

static

Definition at line 4212 of file LinalgOps.cpp.

References build(), and mlir::OpBuilder::create().

create() [2/3]

BatchMatmulTransposeAOp mlir::linalg::BatchMatmulTransposeAOp::create ( OpBuilder & builder, Location location, TypeRange resultTensorTypes, ValueRange inputs, ValueRange outputs, Attribute cast, ArrayRef< NamedAttribute > attributes = {} )

static

Definition at line 4234 of file LinalgOps.cpp.

References build(), and mlir::OpBuilder::create().

create() [3/3]

BatchMatmulTransposeAOp mlir::linalg::BatchMatmulTransposeAOp::create ( OpBuilder & builder, Location location, ValueRange inputs, ValueRange outputs, ArrayRef< NamedAttribute > attributes = {} )

static

Definition at line 4192 of file LinalgOps.cpp.

References build(), and mlir::OpBuilder::create().

Referenced by mlir::linalg::transposeBatchMatmul().

isDefaultIndexingMaps()

bool mlir::linalg::BatchMatmulTransposeAOp::isDefaultIndexingMaps ( Attribute attr )

static

Checks if the affine map is the expected one for this operation.

Definition at line 4169 of file LinalgOps.cpp.

References ArrayAttr(), and mlir::linalg::getAffineResultPositions().

Referenced by classof().

resolveTypeID()

::mlir::TypeID mlir::linalg::BatchMatmulTransposeAOp::resolveTypeID ( )

inlinestatic

Definition at line 258 of file Linalg.h.

References mlir::TypeID::get().

---

The documentation for this class was generated from the following files:

• include/mlir/Dialect/Linalg/IR/Linalg.h
• lib/Dialect/Linalg/IR/LinalgOps.cpp