mlir::linalg::BatchMatmulTransposeBOp Class Reference

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

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

Inheritance diagram for mlir::linalg::BatchMatmulTransposeBOp:

Static Public Member Functions
::mlir::TypeID	resolveTypeID ()
static void	build (OpBuilder &builder, OperationState &result, ValueRange inputs, ValueRange outputs, ArrayRef< NamedAttribute > attributes={})
	Build a transpose B matmul. More...
static BatchMatmulTransposeBOp	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 B matmul with a specific result type. More...
static BatchMatmulTransposeBOp	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 B matmul with a specific result type and a cast type. More...
static BatchMatmulTransposeBOp	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 B.

Definition at line 298 of file Linalg.h.

Member Function Documentation

build() [1/3]

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

static

Build a transpose B matmul with a specific result type.

Definition at line 4284 of file LinalgOps.cpp.

References buildMatmulOp().

build() [2/3]

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

static

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

Definition at line 4305 of file LinalgOps.cpp.

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

build() [3/3]

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

static

Build a transpose B matmul.

Definition at line 4265 of file LinalgOps.cpp.

References buildMatmulOp().

classof()

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

static

Definition at line 4327 of file LinalgOps.cpp.

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

create() [1/3]

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

static

Definition at line 4294 of file LinalgOps.cpp.

create() [2/3]

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

static

Definition at line 4316 of file LinalgOps.cpp.

create() [3/3]

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

static

Definition at line 4274 of file LinalgOps.cpp.

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

isDefaultIndexingMaps()

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

static

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

Definition at line 4251 of file LinalgOps.cpp.

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

Referenced by classof().

resolveTypeID()

::mlir::TypeID mlir::linalg::BatchMatmulTransposeBOp::resolveTypeID ( )

inlinestatic

Definition at line 305 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