Fix Torch check Context

This was causing the Torch option to show up on some
unsupposeted devices.

Change-Id: I750dce1b4d46e08b6fc93a1d9834fd4542669105