From fa43065cbb1a9b47e1283538a51a290be0074cb5 Mon Sep 17 00:00:00 2001
From: Mikhail Treskin <mikhail.treskin@intel.com>
Date: Wed, 12 Aug 2020 18:14:27 +0300
Subject: [PATCH] Fix select references invocation from evaluate_node()

---
 ngraph/test/runtime/interpreter/evaluates_map.cpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/ngraph/test/runtime/interpreter/evaluates_map.cpp b/ngraph/test/runtime/interpreter/evaluates_map.cpp
index b2b216f284531c..a30dab8f4568bb 100644
--- a/ngraph/test/runtime/interpreter/evaluates_map.cpp
+++ b/ngraph/test/runtime/interpreter/evaluates_map.cpp
@@ -352,12 +352,14 @@ namespace {
     bool evaluate(const shared_ptr<op::v1::Select> &op, const HostTensorVector &outputs,
                   const HostTensorVector &input) {
         using T = typename element_type_traits<ET>::value_type;
-        size_t element_count = shape_size(op->get_output_shape(0));
         runtime::reference::select<T>(input[0]->get_data_ptr<const char>(),
                                       input[1]->get_data_ptr<const T>(),
                                       input[2]->get_data_ptr<const T>(),
                                       outputs[0]->get_data_ptr<T>(),
-                                      element_count);
+                                      op->get_input_shape(0),
+                                      op->get_input_shape(1),
+                                      op->get_input_shape(2),
+                                      op->get_auto_broadcast());
         return true;
     }