eager execution and gradients

Question

this is model architecture. I tried to visualize CNN activations but faced with the next error:

RuntimeError: tf.gradients is not supported when eager ex