message: Improve error message during TFR op generation
diff: @@ -99,7 +99,9 @@ class OpRegGenImpl(transformer.CodeGenerator):
  all_func_args = self.visit(node.args)
  if len(expected_args) != len(all_func_args):
  - raise KeyError('Composition arguments do not match the registration.')
  + raise KeyError(
  + 'Composition arguments for {} do not match the registration. {} vs {}'
  + .format(op_name, expected_args, all_func_args))
  cxx_reg_code = ['\nREGISTER_OP("{}")'.format(op_name)]
  for input_ in inputs:
  