From bfb1c7d570e9a783d3b27e79ca88896e685cfae9 Mon Sep 17 00:00:00 2001 From: Evgeny Kotov Date: Thu, 2 Sep 2021 17:09:34 +0300 Subject: [PATCH] use 1D convolution (#7291) --- .../gna_split_convolution_with_large_buffer_size.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference-engine/tests/unit/gna/ngraph/transformations/gna_split_convolution_with_large_buffer_size.cpp b/inference-engine/tests/unit/gna/ngraph/transformations/gna_split_convolution_with_large_buffer_size.cpp index 29dbf117cb40c3..b13d3518286061 100644 --- a/inference-engine/tests/unit/gna/ngraph/transformations/gna_split_convolution_with_large_buffer_size.cpp +++ b/inference-engine/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: