_lib_dir = os.path.dirname(gen_composite_ops.__file__)
_lib_name = os.path.basename(gen_composite_ops.__file__)[4:].replace(
    '.py', '.so')
load_library.load_op_library(os.path.join(_lib_dir, _lib_name))


class GraphDecomposeTest(test.TestCase):

  def testAddN(self):
    add = def_function.function(gen_composite_ops.my_add_n)
    t1 = constant_op.constant([[1.0, 2.0], [3.0, 4.0]])
    t2 = constant_op.constant([[1.0, 2.0], [3.0, 4.0]])

        "//tensorflow/core/common_runtime:device_set",
        "@llvm-project//mlir:IR",
    ],
    alwayslink = 1,
)

tf_py_strict_test(
    name = "graph_decompose_test",
    size = "small",
    srcs = ["integration/graph_decompose_test.py"],
    data = ["//tensorflow/compiler/mlir/tfr/resources:decomposition_lib"],
    python_version = "PY3",
    srcs_version = "PY3",
    tags = [