Test Case: use expected shape, not computed output shape (#3064)

ca220f7d · Michał Karzyński · Scott Cyphers · e21db881 · ca220f7d
Commit ca220f7d authored Jun 17, 2019 by Michał Karzyński Committed by Scott Cyphers Jun 17, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

test_case.hpp test/util/test_case.hpp +1 -2

No files found.
--- a/test/util/test_case.hpp
+++ b/test/util/test_case.hpp
@@ -97,9 +97,8 @@ namespace ngraph
                             "All function results already have expected outputs.");

                auto function_output_type = results.at(m_output_index)->get_element_type();
-                auto function_output_shape = results.at(m_output_index)->get_shape();
                m_result_tensors.emplace_back(
-                    m_backend->create_tensor(function_output_type, function_output_shape));
+                    m_backend->create_tensor(function_output_type, expected_shape));

                m_expected_outputs.emplace_back(std::make_shared<ngraph::op::Constant>(
                    function_output_type, expected_shape, values));