Remove mlir::Type::isa() usages.

This is removed in MLIR upstream:
llvm/llvm-project#135556

PiperOrigin-RevId: 748822888

Author: wecing

tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h

@@ -226,7 +226,8 @@ struct ConvertStatsToQDQs : public OpRewritePattern<quantfork::StatisticsOp> {
       // Per axis quantization (or per channel quantization)
       int stats_num = op.getAxisStats()->getNumElements();
       if (stats_num == 0 || stats_num % 2 != 0) return failure();
-      auto stats = op.getAxisStats()->dyn_cast<DenseFPElementsAttr>();
+      auto stats =
+          llvm::dyn_cast<DenseFPElementsAttr>(op.getAxisStats().value());
       if (!stats) return failure();
 
       for (auto it = stats.begin(), e = stats.end(); it != e; ++it) {
@@ -671,9 +672,7 @@ class QuantizationPattern : public RewritePattern {
   void RewireFloatModelBackbone(Operation* quantized_op,
                                 Operation* float_op) const {
     for (int i = 0, e = quantized_op->getNumResults(); i < e; ++i) {
-      if (!float_op->getResult(i)
-               .getType()
-               .cast<ShapedType>()
+      if (!llvm::cast<ShapedType>(float_op->getResult(i).getType())
                .getElementType()
                .isF32()) {
         continue;

tensorflow/compiler/mlir/tensorflow/ir/tf_executor_ops.td

@@ -47,11 +47,11 @@ def TfExecutor_Dialect : Dialect {
 }
 
 // Control type.
-def TfeControlType : Type<CPred<"$_self.isa<ControlType>()">, "control">,
+def TfeControlType : Type<CPred<"llvm::isa<ControlType>($_self)">, "control">,
                      BuildableType<"$_builder.getType<ControlType>()">;
 
 // Token type.
-def TfeTokenType : Type<CPred<"$_self.isa<TokenType>()">, "token">,
+def TfeTokenType : Type<CPred<"llvm::isa<TokenType>($_self)">, "token">,
                    BuildableType<"$_builder.getType<TokenType>()">;
 
 // TODO(hinsu): Define and use TensorType instead of AnyType for data operands

tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

@@ -561,7 +561,7 @@ LogicalResult ParseExampleV2Op::verify() {
 template <typename CallOpClass>
 static LogicalResult VerifyPartitionedCall(CallOpClass op,
                                            SymbolTableCollection &symbolTable) {
-  SymbolRefAttr func = op->getAttr("f").template cast<SymbolRefAttr>();
+  SymbolRefAttr func = llvm::cast<SymbolRefAttr>(op->getAttr("f"));
   auto function = symbolTable.lookupNearestSymbolFrom<func::FuncOp>(op, func);
   if (!function) {
     return op.emitError("'f' attribute refers to an undefined function: ")

tensorflow/compiler/mlir/tensorflow/utils/convert_type.cc

@@ -180,10 +180,10 @@ absl::Status ConvertScalarTypeToDataType(Type type, DataType* dtype) {
         absl::StrCat("Converting ", debugString(type), " to DataType"));
   }
 
-#define HANDLE_TF_TYPE(tftype, enumerant, name)  \
-  if (type.isa<mlir::tf_type::tftype##Type>()) { \
-    *dtype = DT_##enumerant;                     \
-    return OkStatus();                           \
+#define HANDLE_TF_TYPE(tftype, enumerant, name)       \
+  if (llvm::isa<mlir::tf_type::tftype##Type>(type)) { \
+    *dtype = DT_##enumerant;                          \
+    return OkStatus();                                \
   }
 // NOLINTNEXTLINE
 #include "tensorflow/compiler/mlir/tensorflow/ir/tf_types.def"

tensorflow/core/ir/importexport/convert_types.cc

@@ -171,7 +171,7 @@ Status ConvertScalarTypeToDataType(Type type, DataType* dtype) {
   }
 
 #define HANDLE_TF_TYPE(tftype, enumerant, name) \
-  if (type.isa<tftype##Type>()) {               \
+  if (llvm::isa<tftype##Type>(type)) {          \
     *dtype = tensorflow::DT_##enumerant;        \
     return ::tensorflow::OkStatus();            \
   }

tensorflow/core/ir/types/dialect.cc

@@ -631,7 +631,7 @@ Type TensorFlowRefType::RemoveRef() {
   if (mlir::isa<Complex128RefType>(*this))
     return ComplexType::get(Float64Type::get(ctx));
 #define HANDLE_TF_TYPE(tftype, enumerant, name) \
-  if (isa<tftype##RefType>()) return tftype##Type::get(ctx);
+  if (mlir::isa<tftype##RefType>(*this)) return tftype##Type::get(ctx);
 
 #define HANDLE_TF_REF_TYPE(tftype, enumerant, name)
 // NOLINTNEXTLINE