Tensor type inference trait that constructs a tensor from the inferred shape and elemental types. More...
#include "mlir/Interfaces/InferTypeOpInterface.h"
Inheritance diagram for mlir::OpTrait::InferTensorType< ConcreteType >:
Additional Inherited Members | |
| Protected Member Functions inherited from mlir::OpTrait::TraitBase< ConcreteType, InferTensorType > | |
| Operation * | getOperation () |
| Return the ultimate Operation being worked on. More... | |
Detailed Description
template<typename ConcreteType>
class mlir::OpTrait::InferTensorType< ConcreteType >
Tensor type inference trait that constructs a tensor from the inferred shape and elemental types.
Requires: Op implements InferShapedTypeOpInterface and InferTypeOpInterface. Less strict is possible (e.g., implements inferReturnTypeComponents and these always populates all element types and shapes or fails, but this trait is currently only used where the interfaces are, so keep it restricted for now).
Definition at line 277 of file InferTypeOpInterface.h.
The documentation for this class was generated from the following file:
- include/mlir/Interfaces/InferTypeOpInterface.h
