fix micro-optim

Author: pranavm-nvidia

tripy/tripy/backend/mlir/executor.py

@@ -108,7 +108,7 @@ def get_output_tensor_runtime_info(self, inputs, output_devices=List[device]):
         outputs_shape, all_outputs_known = self._get_outputs_shape()
         if not all_outputs_known:
             inputs_shape = self._get_inputs_runtime_shape(inputs)
-            outputs_shape = self._execute_shape_inference(inputs_shape)
+            outputs_shape = self._execute_shape_inference(inputs_shape, outputs_shape)
         output_tensor_info = self._get_output_tensor_info(outputs_shape, output_devices)
         return output_tensor_info