I assume the window in the top left is showing the utilization?
For other scripts it’s showing a high usage?
Could you try to profile your code with e.g. utils.bottleneck 105.
Also, could you create random inputs with your shapes, so that I could try it on my machine?