We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 53f0812 commit 9a8cc6cCopy full SHA for 9a8cc6c
src/plugins/intel_gpu/src/graph/impls/onednn/fully_connected_onednn.cpp
@@ -160,6 +160,13 @@ struct fully_connected_onednn : typed_primitive_onednn_impl<fully_connected> {
160
if (prim_input_size < 4) {
161
auto output_pshape = output_layout.get_partial_shape();
162
if (output_pshape.size() > prim_input_size) {
163
+ std::vector<ov::Dimension> new_dims;
164
+ for (const auto& dim : output_pshape) {
165
+ if (!dim.is_static() || dim.get_length() != 1)
166
+ new_dims.push_back(dim);
167
+ }
168
+ ov::PartialShape new_shape(new_dims);
169
+ output_pshape = new_shape;
170
output_pshape.resize(prim_input_size);
171
output_layout.set_partial_shape(output_pshape);
172
}
0 commit comments