Hi,
You will need to give more background on the hardware you’re using to get a better idea.
Also this is a third party implementation so it might not be very well optimized. compared to the builtin version provided by tensorflow.