How to obtain the gradient of the categorical cross entropy w.r.t. the input image in Tensorflow.js?

Question

I\'m trying to obtain the gradient of the loss objective, in my case categorical cross entropy w.r.t to the input image in Tensorflow.js.

I\'ve tried this so far: