proper shape for 2 outputs

michal-miotk · Jul 25, 2024 · c00ff8a · c00ff8a
1 parent fae772a
commit c00ff8a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/plugins/intel_gpu/src/graph/impls/ocl/lstm_seq.cpp b/src/plugins/intel_gpu/src/graph/impls/ocl/lstm_seq.cpp
@@ -74,6 +74,9 @@ struct lstm_seq_impl : typed_primitive_impl_ocl<lstm_seq> {
         params.clip = primitive->clip;
         params.input_forget = primitive->input_forget;
         params.direction = primitive->direction;
+        //Legacy multi-output
+        params.outputs.push_back(convert_data_tensor(impl_param.input_layouts[1]));
+        params.outputs.push_back(convert_data_tensor(impl_param.input_layouts[1]));
         params.outputs.push_back(convert_data_tensor(impl_param.get_output_layout(1)));
         params.outputs.push_back(convert_data_tensor(impl_param.get_output_layout(2)));