19 lines
508 B
TableGen
19 lines
508 B
TableGen
#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
|