Tensorflow: Checking condition in call method of custom layer using tf.cond()

Question

I\'m implementing a custom layer in tensorflow 2.x . My requirement is such that, the program should check a condition before returning the output.

class SimpleRN