message: In accordance to changes in get_concrete_function.
diff: @@ -1103,9 +1103,6 @@ def register(func, *args, **kwargs):
  function definition into graph. Register function with different input param
  will result into multiple version of functions registered in graph.
  - Also, `args` and `kwargs` are ignored if this `PolymorphicFunction` was
  - created with an `input_signature`.
  -
  Args:
  func: the PolymorphicFunction instance that generated by a @defun
  *args: input arguments for the Python function.
  