#include <LinalgTraits.h>
Inheritance diagram for mlir::OpTrait::linalg::StructuredOpTraits< ConcreteType >:
Collaboration diagram for mlir::OpTrait::linalg::StructuredOpTraits< ConcreteType >:
Public Member Functions | |
| Value | getInput (unsigned i) |
Return the i-th input value. More... | |
| Optional< unsigned > | getIndexOfInput (Value value) |
| ShapedType | getInputShapedType (unsigned i) |
Return the i-th input buffer type. More... | |
| Operation::operand_range | getInputs () |
| Return the range over inputs. More... | |
| Value | getOutput (unsigned i) |
Return the i-th output. More... | |
| Optional< unsigned > | getIndexOfOutput (Value value) |
| ShapedType | getOutputShapedType (unsigned i) |
Return the i-th output buffer type. More... | |
| bool | hasBufferSemantics () |
| Query whether the op has only MemRef input and outputs. More... | |
| SmallVector< RankedTensorType, 4 > | getInputTensorTypes () |
| Query the subset of input operands that are of ranked tensor type. More... | |
| SmallVector< RankedTensorType, 4 > | getOutputTensorTypes () |
| Query the subset of output operands that are of ranked tensor type. More... | |
| Operation::operand_range | getOutputs () |
| Return the range over outputs. More... | |
| unsigned | getNumInputsAndOutputs () |
| Return the number of inputs and outputs. More... | |
| ShapedType | getShapedType (unsigned i) |
Return the i-th buffer type. More... | |
| Operation::operand_range | getInputsAndOutputs () |
| Return the range over inputs and outputs. More... | |
| unsigned | getNumParallelLoops () |
| unsigned | getNumReductionLoops () |
| unsigned | getNumWindowLoops () |
| unsigned | getNumLoops () |
Static Public Member Functions | |
| static LogicalResult | verifyTrait (Operation *op) |
Additional Inherited Members | |
 Protected Member Functions inherited from mlir::OpTrait::TraitBase< ConcreteType, StructuredOpTraits > | |
| Operation * | getOperation () |
| Return the ultimate Operation being worked on. More... | |
| Static Protected Member Functions inherited from mlir::OpTrait::TraitBase< ConcreteType, StructuredOpTraits > | |
| static LogicalResult | verifyTrait (Operation *op) |
| static AbstractOperation::OperationProperties | getTraitProperties () |
Detailed Description
template<typename ConcreteType>
class mlir::OpTrait::linalg::StructuredOpTraits< ConcreteType >
This class provides the API for structured ops that are known to operate on buffers or tensors. This trait must be used in conjunction with an op definition or a trait that provides the methods getNumInputs and getNumOutputs. Use as a trait as follows:
class DotOp : public Op<DotOp, OpTrait::StructuredOpTraits> {
Member Function Documentation
◆ getIndexOfInput()
template<typename ConcreteType >
|
inline |
Return the index of value in the list of inputs if found, llvm::None otherwise.
◆ getIndexOfOutput()
template<typename ConcreteType >
|
inline |
Return the index of value in the list of output values if found, llvm::None otherwise.
◆ getInput()
template<typename ConcreteType >
|
inline |
Return the i-th input value.
◆ getInputs()
template<typename ConcreteType >
|
inline |
Return the range over inputs.
◆ getInputsAndOutputs()
template<typename ConcreteType >
|
inline |
Return the range over inputs and outputs.
◆ getInputShapedType()
template<typename ConcreteType >
|
inline |
Return the i-th input buffer type.
◆ getInputTensorTypes()
template<typename ConcreteType >
|
inline |
Query the subset of input operands that are of ranked tensor type.
◆ getNumInputsAndOutputs()
template<typename ConcreteType >
|
inline |
Return the number of inputs and outputs.
◆ getNumLoops()
template<typename ConcreteType >
|
inline |
◆ getNumParallelLoops()
template<typename ConcreteType >
|
inline |
◆ getNumReductionLoops()
template<typename ConcreteType >
|
inline |
◆ getNumWindowLoops()
template<typename ConcreteType >
|
inline |
◆ getOutput()
template<typename ConcreteType >
|
inline |
Return the i-th output.
◆ getOutputs()
template<typename ConcreteType >
|
inline |
Return the range over outputs.
◆ getOutputShapedType()
template<typename ConcreteType >
|
inline |
Return the i-th output buffer type.
◆ getOutputTensorTypes()
template<typename ConcreteType >
|
inline |
Query the subset of output operands that are of ranked tensor type.
◆ getShapedType()
template<typename ConcreteType >
|
inline |
Return the i-th buffer type.
◆ hasBufferSemantics()
template<typename ConcreteType >
|
inline |
Query whether the op has only MemRef input and outputs.
◆ verifyTrait()
template<typename ConcreteType >
|
inlinestatic |
The documentation for this class was generated from the following file:
- include/mlir/Dialect/Linalg/IR/LinalgTraits.h
