EXPECT_EQ(
      compilation_status.Delta(mlir::TF::kMlirPh1BridgeCounterNonReplicated,
                               mlir::TF::kMlirPh1BridgeCounterV2,
                               mlir::TF::kMlirPh1BridgeCounterNonTpu,
                               "fallback_disabled", "success"),
      1);
}

TEST_F(FunctionClusterTensorflowDialectTest, LogsFallbackMode) {
  CellReader<int64_t> compilation_status(kCompilationStreamz);

  std::string device_type = is_supported_by_replicated_brige
                                ? mlir::TF::kMlirPh1BridgeCounterTpu
                                : mlir::TF::kMlirPh1BridgeCounterNonTpu;

  VLOG(2)
      << (is_supported_by_replicated_brige ? "Replicated" : "NonReplicated")
      << " Bridge called stack trace is "

inline const char kMlirPh1BridgeCounterV1[] = "v1";
inline const char kMlirPh1BridgeCounterV2[] = "v2";
inline const char kMlirPh1BridgeCounterTpu[] = "tpu";
inline const char kMlirPh1BridgeCounterNonTpu[] = "cpu/gpu";
inline const char kXlaOutsideCompilation[] = "_xla_outside_compilation";

// Copies attributes that satisfy the given predicate from `from` to `to`.
template <typename Predicate>