_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]])