| 296 | def test_docstring(self): |
| 297 | |
| 298 | class Adder(module.Module): |
| 299 | |
| 300 | @def_function.function(input_signature=[tensor_spec.TensorSpec( |
| 301 | shape=None, dtype=dtypes.float32)]) |
| 302 | def add(self, x): |
| 303 | return x + x + 1. |
| 304 | |
| 305 | to_save = Adder() |
| 306 | to_save.add(constant_op.constant(1.)) |