cc_test(
    name = "beam_benchmark",
    srcs = ["beam_benchmark.cc"],
    deps = [
        "//dragnn/core:beam",
        "//dragnn/core/interfaces:cloneable_transition_state",
        "//dragnn/core/interfaces:transition_state",
        "//dragnn/core/test:mock_transition_state",
        "@org_tensorflow//tensorflow/core:test",
    ],
)