use 1D convolution (#7291)

openvinotoolkit · Sep 2, 2021 · 6cbeb18 · 6cbeb18
1 parent 90b2265
commit 6cbeb18
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...ne/tests/unit/gna/ngraph/transformations/gna_split_convolution_with_large_buffer_size.cpp b/...ne/tests/unit/gna/ngraph/transformations/gna_split_convolution_with_large_buffer_size.cpp
@@ -78,7 +78,7 @@ class CreateAppendableGraphDecorator : public CreateGraphDecorator {
 class CreateBaseDecorator : public CreateGraphDecorator {
 public:
     // always the first decorator => no prev_builder
-    CreateBaseDecorator(const ngraph::Shape& input_data_shape = ngraph::Shape{1, 64, 4096, 4096}) :
+    CreateBaseDecorator(const ngraph::Shape& input_data_shape = ngraph::Shape{1, 64, 1, 4096}) :
                         CreateGraphDecorator(nullptr),
                         input_data_shape_(input_data_shape) {}
 protected: