Updating the type for the inputs to the select layer

Author: apbose

py/torch_tensorrt/dynamo/conversion/impl/condition/ops.py

@@ -104,9 +104,9 @@ def select(
     target: Target,
     source_ir: Optional[SourceIR],
     name: str,
-    input: Union[TRTTensor, np.ndarray, torch.Tensor],
-    other: Union[TRTTensor, np.ndarray, torch.Tensor],
-    condition: Union[TRTTensor, np.ndarray, torch.Tensor],
+    input: Union[TRTTensor],
+    other: Union[TRTTensor],
+    condition: Union[TRTTensor],
 ) -> TRTTensor:
     select_layer = ctx.net.add_select(condition, input, other)
     set_layer_name(select_layer, target, name + "_select", source_ir)