(self)
| 1023 | nn_ops.softplus, features, expected=expected, rtol=1e-6, atol=9.1e-6) |
| 1024 | |
| 1025 | def testSoftplus(self): |
| 1026 | for dtype in self.float_types: |
| 1027 | self._assertSoftplusMatchesExpected([[-2, 0, 8]], dtype) |
| 1028 | self._assertSoftplusMatchesExpected( |
| 1029 | [[-9, 7, -5, 3, -1], [1, -3, 5, -7, 9]], dtype) |
| 1030 | if dtype == dtypes.bfloat16.as_numpy_dtype: |
| 1031 | log_eps = np.log(np.finfo(np.float32).eps) |
| 1032 | else: |
| 1033 | log_eps = np.log(np.finfo(dtype).eps) |
| 1034 | one = dtype(1) |
| 1035 | ten = dtype(10) |
| 1036 | self._assertSoftplusMatchesExpected([ |
| 1037 | log_eps, log_eps - one, log_eps + one, log_eps - ten, log_eps + ten, |
| 1038 | -log_eps, -log_eps - one, -log_eps + one, -log_eps - ten, |
| 1039 | -log_eps + ten |
| 1040 | ], dtype) |
| 1041 | |
| 1042 | |
| 1043 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected