I am trying to process gradients in a Keras layer before applying them as stated in
https://www.tensorflow.org/api_docs/python/tf/keras/optimizers/Optimizer
with t