}

// Control type.
def TfeControlType : Type<CPred<"$_self.isa<ControlType>()">, "control">,
                     BuildableType<"$_builder.getType<ControlType>()">;

// Token type.
def TfeTokenType : Type<CPred<"$_self.isa<TokenType>()">, "token">,
                   BuildableType<"$_builder.getType<TokenType>()">;

// TODO(hinsu): Define and use TensorType instead of AnyType for data operands

// tensor argument types
class TFR_Type<string name> : DialectType<TFR_Dialect,
    CPred<"$_self.isa<mlir::TFR::" # name # "Type>()">,
    "TFR " # name #" type">,
    BuildableType<"$_builder.getType<mlir::TFR::" # name # "Type>()">;
def TFR_TensorType : TFR_Type<"TFRTensor">;
def TFR_TensorListType : TFR_Type<"TFRTensorList">;
def TFR_AllTensorTypes : Type<Or<[
    TFR_TensorType.predicate,

// Class for any TensorFlow dialect specific type
class TF_TensorFlowType <string name, string description> :
    Type<CPred<"$_self.isa<mlir::TF::" # name # "Type>()">,
         "TensorFlow " # description # " type">,
    BuildableType<"getType<mlir::TF::" # name # "Type>()">;

//===----------------------------------------------------------------------===//
// Reference types

// Float reference types