use 1D convolution

openvinotoolkit · Aug 30, 2021 · ef149be · ef149be
1 parent 4a9ac22
commit ef149be
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: