I have a TensorFlow model converted to TensorFlow Lite, where my objective is to run it on Raspberry Pi. My model is a classification model that uses quantization aware training before conversion to the TFLite model. According to this example: Grad-CAM class activation visualization 6. I successfully applied grad-cam to an actual Tensorflow model. However, I did not find the methods to compute the gradients for Tensorflow Lite.
How can I replicate the TensorFlow grad-cam but for TensorFlow lite?