xref: /llvm-project/mlir/include/mlir/Dialect/Tensor/TransformOps/TensorTransformOps.h (revision 5a9bdd85ee4d8527e2cedf44f3ce26ff414f9b6a) 
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 &registry);
22 void registerFindPayloadReplacementOpInterfaceExternalModels(
23     DialectRegistry &registry);
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