Tensorflow tf.function conditionals

Question

It took me some time to pin down the problem. Here is is:

class ComplicatedStuff:
    def __init__(self):
        self.resu