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. More...
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. More...
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. More...
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. More...
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::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 4191 of file LinalgOps.cpp.

References buildMatmulOp().

build() [2/3]

void mlir::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 4212 of file LinalgOps.cpp.

References mlir::OperationState::addAttribute(), and buildMatmulOp().

build() [3/3]

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

static

Build a transpose A matmul.

Definition at line 4172 of file LinalgOps.cpp.

References buildMatmulOp().

classof()

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

static

Definition at line 4234 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 4201 of file LinalgOps.cpp.

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 4223 of file LinalgOps.cpp.

create() [3/3]

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

static

Definition at line 4181 of file LinalgOps.cpp.

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 4158 of file LinalgOps.cpp.

References mlir::remark::failed(), and mlir::linalg::getAffineResultPositions().

Referenced by classof().

resolveTypeID()

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

inlinestatic

Definition at line 258 of file Linalg.h.

---

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

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