feat: toy tutorial chapter 4.
Signed-off-by: jackfiled <xcrenchangjun@outlook.com>
This commit is contained in:
18
include/hello/ShapeInferenceInterface.td
Normal file
18
include/hello/ShapeInferenceInterface.td
Normal file
@@ -0,0 +1,18 @@
|
||||
#ifndef SHAPE_INFERENCE_INTERFACE
|
||||
#define SHAPE_INFERENCE_INTERFACE
|
||||
|
||||
include "mlir/IR/OpBase.td"
|
||||
|
||||
def ShapeInferenceOpInterface : OpInterface<"ShapeInference"> {
|
||||
let description = [{
|
||||
Interface to access a registered method to infer the return types for an
|
||||
operation that can be used during type inference.
|
||||
}];
|
||||
|
||||
let methods = [
|
||||
InterfaceMethod<"Infer and set the output shape for the current operation.",
|
||||
"void", "inferShapes">
|
||||
];
|
||||
}
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user