Add `torch.inference_session` on runner function

[Anindyadeep]

There are two ways of doing this,

1. either put a decorator on the base class function or inside each of the method
2. Or use within the method using with