message: [NFC] Add ops to use MLIR legalization when prefer_tf2xla is on. This can be the initial opset to start testing prefer_tf2xla as default. Ops added are those that use MlirXlaOpKernel in the old bridge. Other ops added have no XlaOpKernel legalization.
diff: @@ -6428,6 +6428,56 @@ const llvm::DenseSet<mlir::TypeID> &MlirPreferredOps() {
  new llvm::DenseSet<mlir::TypeID>{
  // Ops that are legalized in the old bridge using MlirXlaOpKernel
  TypeID::get<TF::AbsOp>(),
  + TypeID::get<TF::AtanOp>(),
  + TypeID::get<TF::AvgPool3DOp>(),
  + TypeID::get<TF::BiasAddGradOp>(),
  + TypeID::get<TF::CeilOp>(),
  + TypeID::get<TF::CheckNumericsOp>(),
  + TypeID::get<TF::ComplexOp>(),
  + TypeID::get<TF::CosOp>(),
  + TypeID::get<TF::DiagPartOp>(),
  + TypeID::get<TF::DivOp>(),
  + TypeID::get<TF::EinsumOp>(),
  + TypeID::get<TF::ExpOp>(),
  + TypeID::get<TF::Expm1Op>(),
  + TypeID::get<TF::FakeQuantWithMinMaxArgsOp>(),
  + TypeID::get<TF::FloorOp>(),
  + TypeID::get<TF::GreaterEqualOp>(),
  + TypeID::get<TF::IFFTOp>(),
  + TypeID::get<TF::ImagOp>(),
  + TypeID::get<TF::IsFiniteOp>(),
  + TypeID::get<TF::IsInfOp>(),
  + TypeID::get<TF::IsNanOp>(),
  + TypeID::get<TF::LessEqualOp>(),
  + TypeID::get<TF::LgammaOp>(),
  + TypeID::get<TF::Log1pOp>(),
  + TypeID::get<TF::LogicalOrOp>(),
  + TypeID::get<TF::LogSoftmaxOp>(),
  + TypeID::get<TF::MatrixBandPartOp>(),
  + TypeID::get<TF::MaxPool3DGradOp>(),
  + TypeID::get<TF::PreventGradientOp>(),
  + TypeID::get<TF::RandomShuffleOp>(),
  + TypeID::get<TF::RealOp>(),
  + TypeID::get<TF::ReciprocalOp>(),
  + TypeID::get<TF::ReluOp>(),
  + TypeID::get<TF::Relu6Op>(),
  + TypeID::get<TF::ReluGradOp>(),
  + TypeID::get<TF::RsqrtOp>(),
  + TypeID::get<TF::SelectOp>(),
  + TypeID::get<TF::SigmoidOp>(),
  + TypeID::get<TF::SignOp>(),
  + TypeID::get<TF::SoftmaxOp>(),
  + TypeID::get<TF::SqrtOp>(),
  + TypeID::get<TF::SqrtGradOp>(),
  + TypeID::get<TF::SquaredDifferenceOp>(),
  + TypeID::get<TF::TanhOp>(),
  + TypeID::get<TF::TanhGradOp>(),
  + TypeID::get<TF::XlogyOp>(),
  + TypeID::get<TF::ZetaOp>(),
  +
  + // Ops that have no XlaOpKernel.
  + TypeID::get<TF::RiscAddOp>(),
  + TypeID::get<TF::RiscDotOp>(),
  };
  // clang-format on
  return *ops;
  