From c00ff8a0bc37935d89ea6f9cd0bc2608b6a27c0f Mon Sep 17 00:00:00 2001 From: michal-miotk Date: Thu, 25 Jul 2024 15:31:45 +0000 Subject: [PATCH] proper shape for 2 outputs --- src/plugins/intel_gpu/src/graph/impls/ocl/lstm_seq.cpp | 3 +++ 1 file changed, 3 insertions(+) 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 index d53586aae22712..01621bd13856a6 100644 --- 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 { 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)));