Thanks @IvanKobzarev it worked.
But you should explicitly put a restriction on no of threads user is setting because after a certain limit instead of decreasing run-time it is increasing.
One more doubt-
Just like in pytorch can we pass as batch of images in case of pytorch android. ?
Thanks,
Mohit Ranawat