CHECK: %Const = tfr.call @tf__const(%{{.*}}, %[[attr]]) : (!tfr.attr, !tfr.attr) -> (!tfr.tensor)
    """
    self._check_code(mlir_code, mlir_code_exp)

  def test_tfr_tf_ops(self):
    mlir_code = tfr_gen(sys.modules[__name__], '_tfr_tf_ops', [test_ops])
    mlir_code_exp = r"""
      CHECK-LABEL: tfr.func @tf__test_complex_tf_op(%lhs: !tfr.tensor, %rhs: !tfr.tensor) -> (!tfr.tensor_list) {