TfSavedModelTest() : ctx_() {
    ctx_.loadDialect<TensorFlowSavedModelDialect, func::FuncDialect>();
  }

  MLIRContext ctx_;
};

// Parses `module_op_str` and returns the resulting `ModuleOp`.
ModuleOp ParseModuleOp(const StringRef module_op_str, Block& block,
                       MLIRContext& ctx) {
  const LogicalResult parse_result =
      parseSourceString(module_op_str, &block, ParserConfig(&ctx));