(self)
| 198 | save.save(root, save_dir) |
| 199 | |
| 200 | def test_nested_inputs(self): |
| 201 | root = tracking.AutoTrackable() |
| 202 | root.f = def_function.function( |
| 203 | lambda x: 2. * x[0], |
| 204 | input_signature=([tensor_spec.TensorSpec(None, dtypes.float32), |
| 205 | tensor_spec.TensorSpec(None, dtypes.float32)],)) |
| 206 | root.f([constant_op.constant(1.), constant_op.constant(1.)]) |
| 207 | |
| 208 | def test_nested_outputs(self): |
| 209 | root = tracking.AutoTrackable() |