Function model_closure

tests/integration/test_dynamic_shapes.py:61–62

()

    @tf.function
    def model_closure() -> tf.Tensor:
        return -model.elbo()

    model_closure()  # Trigger compilation.