Method func

tensorflow/python/kernel_tests/cwise_ops_unary_test.py:178–185

(x)

  def _replace_domain_error_with_inf(self, fn):

    def func(x):
      try:
        return fn(x)
      except ValueError as e:
        if "domain error" in str(e):
          return np.inf * np.ones_like(x)
        else:
          raise e

    return func