From 930105e1d85a95efb9f9c91987e6d143beee1ce4 Mon Sep 17 00:00:00 2001 From: Evgeny Kotov Date: Mon, 30 Aug 2021 16:14:30 +0300 Subject: [PATCH] use 1D convolution --- .../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: