Hello @vishalagarwal ,
Which version of pytorch android api do you use 1.3, 1.3.1 or nightly builds?
We had a deadlock issue that was fixed in  https://github.com/pytorch/pytorch/pull/29885 2
If you are not using nightly builds, could you please check if it reproduces on nightlies?(https://github.com/pytorch/pytorch/tree/master/android#nightly 4)