chore: remove device placement of input tensors

Signed-off-by: Dheeraj Peri <[email protected]>

Author: peri044

py/torch_tensorrt/fx/passes/pass_utils.py

@@ -139,9 +139,7 @@ def pass_with_validation(
             *args,
             **kwargs,
         ) -> fx.GraphModule:
-            input_tensors = extract_example_tensors_from_input(
-                input, next(module.parameters()).device
-            )
+            input_tensors = extract_example_tensors_from_input(input)
             res0 = module(*input_tensors)
             processed_module = pass_(module, input, *args, **kwargs)
             res1 = processed_module(*input_tensors)