| 240 | f_pruned = f_wrapped.prune(x_in[0], [x_out[0]]) |
| 241 | |
| 242 | class Adder(module.Module): |
| 243 | |
| 244 | @def_function.function(input_signature=[ |
| 245 | tensor_spec.TensorSpec(shape=None, dtype=dtypes.float32)]) |
| 246 | def add(self, x): |
| 247 | return f_pruned(x) |
| 248 | |
| 249 | root = Adder() |
| 250 | root.add(constant_op.constant(1.)) |
no outgoing calls