"/tensorflow/core/tf2xla/api/v2/dynamism_function_counter",
    "Counts how many functions are dynamic", "has_dynamism");

constexpr char kNotDynamicFunctionName[] = "kNotDynamicFunction";
constexpr char kDynamicFunctionName[] = "kDynamicFunction";

class InputMetricsLoweringPass
    : public impl::InputLoweringMetricsPassBase<InputMetricsLoweringPass> {
 public:
  void runOnOperation() override;

namespace {

using ::mlir::LogicalResult;
using ::mlir::ModuleOp;
using ::mlir::mhlo::test::GetMlirModuleFromString;
using ::tensorflow::monitoring::testing::CellReader;

constexpr char kNotDynamicFunctionName[] = "kNotDynamicFunction";
constexpr char kDynamicFunctionName[] = "kDynamicFunction";
static constexpr char kDynamismOpCounterStreamzName[] =
    "/tensorflow/core/tf2xla/api/v2/dynamism_op_counter";

TEST(LegalizeWithCombinedBridge, RecordsDynamicOps) {
  static constexpr char kDynamismFunctionCounterStreamzName[] =
      "/tensorflow/core/tf2xla/api/v2/dynamism_function_counter";
  constexpr char kNotDynamicFunctionName[] = "kNotDynamicFunction";
  CellReader<int64_t> dynamic_function_op_count(
      kDynamismFunctionCounterStreamzName);

  auto result = CompileMlirModule(kMlirModuleStr);

  ASSERT_TRUE(result.ok());