1 /* Copyright 2019 The TensorFlow Authors. All Rights Reserved. 2 3 Licensed under the Apache License, Version 2.0 (the "License"); 4 you may not use this file except in compliance with the License. 5 You may obtain a copy of the License at 6 7 http://www.apache.org/licenses/LICENSE-2.0 8 9 Unless required by applicable law or agreed to in writing, software 10 distributed under the License is distributed on an "AS IS" BASIS, 11 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 See the License for the specific language governing permissions and 13 limitations under the License. 14 ==============================================================================*/ 15 16 // This file defines the types used in the standard MLIR TensorFlow dialect. 17 18 #ifndef TENSORFLOW_COMPILER_MLIR_TENSORFLOW_IR_TF_TYPES_H_ 19 #define TENSORFLOW_COMPILER_MLIR_TENSORFLOW_IR_TF_TYPES_H_ 20 21 #include "tensorflow/core/ir/types/dialect.h" 22 23 namespace mlir { 24 namespace TF { 25 26 // This all moved under tensorflow/core/ir/types and these using declaration are 27 // to help with the transition. 28 29 using ::mlir::tf_type::AreCastCompatible; // NOLINT 30 using ::mlir::tf_type::ArraysAreCastCompatible; // NOLINT 31 using ::mlir::tf_type::BroadcastCompatible; // NOLINT 32 using ::mlir::tf_type::DropRefType; // NOLINT 33 using ::mlir::tf_type::filter_resources; // NOLINT 34 using ::mlir::tf_type::GetCastCompatibleType; // NOLINT 35 using ::mlir::tf_type::HasCompatibleElementTypes; // NOLINT 36 using ::mlir::tf_type::IsValidTFTensorType; // NOLINT 37 using ::mlir::tf_type::OperandShapeIterator; // NOLINT 38 using ::mlir::tf_type::ResourceType; // NOLINT 39 using ::mlir::tf_type::ResultShapeIterator; // NOLINT 40 using ::mlir::tf_type::ResultShapeRange; // NOLINT 41 using ::mlir::tf_type::StringType; // NOLINT 42 using ::mlir::tf_type::TensorFlowRefType; // NOLINT 43 using ::mlir::tf_type::TensorFlowType; // NOLINT 44 using ::mlir::tf_type::TensorFlowTypeWithSubtype; // NOLINT 45 using ::mlir::tf_type::VariantType; // NOLINT 46 47 #define HANDLE_TF_TYPE(tftype, enumerant, name) \ 48 using tftype##Type = mlir::tf_type::tftype##Type; 49 #include "tensorflow/compiler/mlir/tensorflow/ir/tf_types.def" 50 51 52 } // end namespace TF 53 } // end namespace mlir 54 55 #endif // TENSORFLOW_COMPILER_MLIR_TENSORFLOW_IR_TF_TYPES_H_ 56