Tensorflow custom gradients don't backpropagate

Question

I\'m trying to understand custom gradients in tensorflow with tf.custom_gradient, so I was trying to reproduce a simple matrix multiplication with its gradients, wh