1 | //===- TensorTransformOps.h - Tensor transformation ops ---------*- C++ -*-===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | |
9 | #ifndef MLIR_DIALECT_TENSOR_TRANSFORMOPS_TENSORTRANSFORMOPS_H |
10 | #define MLIR_DIALECT_TENSOR_TRANSFORMOPS_TENSORTRANSFORMOPS_H |
11 | |
12 | #include "mlir/Dialect/Transform/IR/TransformTypes.h" |
13 | #include "mlir/Dialect/Transform/Interfaces/TransformInterfaces.h" |
14 | #include "mlir/IR/OpImplementation.h" |
15 | #include "mlir/IR/PatternMatch.h" |
16 | |
17 | namespace mlir { |
18 | class DialectRegistry; |
19 | |
20 | namespace tensor { |
21 | void registerTransformDialectExtension(DialectRegistry ®istry); |
22 | void registerFindPayloadReplacementOpInterfaceExternalModels( |
23 | DialectRegistry ®istry); |
24 | } // namespace tensor |
25 | } // namespace mlir |
26 | |
27 | #define GET_OP_CLASSES |
28 | #include "mlir/Dialect/Tensor/TransformOps/TensorTransformOps.h.inc" |
29 | |
30 | #endif // MLIR_DIALECT_TENSOR_TRANSFORMOPS_TENSORTRANSFORMOPS_H |
31 | |