aten::where mypy changes- cases where x_shape!=y_shape. ToDO: to add more such test cases

Author: apbose

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

@@ -65,15 +65,15 @@ def where(
         condition_val = condition_layer.get_output(0)
     else:
         assert condition.dtype == trt.bool, "mask dtype is not bool!"
-        if condition_shape != condition_dim:  # TODO: What is this checking?
+        if len(condition_shape) != condition_dim:  # TODO: What is this checking?
             condition_val = expand(
                 network, target, source_ir, f"{name}_expand", condition, output_shape
             )
         else:
             condition_val = condition
 
     if type(input) != TRTTensor:
-        if x_shape != input_dim:  # TODO: What is this checking?
+        if x_shape != output_shape:  # TODO: What is this checking?
             # special case where 1 element in input
             if len(input.shape) == 0:
                 input = input.unsqueeze(0)
@@ -95,7 +95,7 @@ def where(
         y_val = get_trt_tensor(network, other, f"{name}_y")
     else:
         y_val = other
-        if y_shape != other_dim:  # TODO: What is this checking?
+        if y_shape != output_shape:  # TODO: What is this checking?
             y_val = expand(
                 network, target, source_ir, f"{name}_y_expand", y_val, output_shape
             )