outputs=['o1: T', 'o2: T'])
def _composite_op(x, y, act, trans):
  return x + act, y + trans


class TFRGenTensorTest(test.TestCase):
  """MLIR Generation Tests for MLIR TFR Program."""

  def test_op_reg_gen(self):
    cxx_code = gen_register_op(sys.modules[__name__])
    cxx_code_exp = r"""
      CHECK: #include "tensorflow/core/framework/op.h"
      CHECK-EMPTY
      CHECK: namespace tensorflow {
      CHECK-EMPTY