Unverified Commit 6a369747 authored by aslepko's avatar aslepko Committed by GitHub

Aslepko/test (#4307)

* Update onnx_import_provenance.in.cpp

* Update onnx_import_dyn_shapes.in.cpp

* Update constant_folding_one_hot.cpp

* Update constant.hpp

* Update constant.hpp

* Update constant_folding_one_hot.cpp

* Update onnx_import_provenance.in.cpp

* Update onnx_import_dyn_shapes.in.cpp
Co-authored-by: 's avatarScott Cyphers <diyessi@users.noreply.github.com>
parent 5a656a1a
...@@ -85,6 +85,7 @@ shared_ptr<op::Constant> fold_constant_one_hot(const shared_ptr<op::Constant>& i ...@@ -85,6 +85,7 @@ shared_ptr<op::Constant> fold_constant_one_hot(const shared_ptr<op::Constant>& i
case element::Type_t::u64: case element::Type_t::u64:
return fold_constant_one_hot_ref<uint64_t, OUTPUT_TYPE>( return fold_constant_one_hot_ref<uint64_t, OUTPUT_TYPE>(
indices, on_value, off_value, output_shape, axis); indices, on_value, off_value, output_shape, axis);
default: NGRAPH_CHECK(false, "Indices input element type must be integer");
} }
} }
......
...@@ -113,7 +113,7 @@ NGRAPH_TEST(onnx_dyn_shapes_${BACKEND_NAME}, scalar_initializers_shape_check) ...@@ -113,7 +113,7 @@ NGRAPH_TEST(onnx_dyn_shapes_${BACKEND_NAME}, scalar_initializers_shape_check)
const auto function = onnx_import::import_onnx_model( const auto function = onnx_import::import_onnx_model(
file_util::path_join(SERIALIZED_ZOO, "onnx/dynamic_shapes/scalar_initializers.prototxt")); file_util::path_join(SERIALIZED_ZOO, "onnx/dynamic_shapes/scalar_initializers.prototxt"));
for (const auto ng_node : function->get_ordered_ops()) for (auto ng_node : function->get_ordered_ops())
{ {
if (as_type_ptr<default_opset::Constant>(ng_node)) if (as_type_ptr<default_opset::Constant>(ng_node))
{ {
......
...@@ -36,9 +36,9 @@ NGRAPH_TEST(onnx_${BACKEND_NAME}, provenance_tag_text) ...@@ -36,9 +36,9 @@ NGRAPH_TEST(onnx_${BACKEND_NAME}, provenance_tag_text)
file_util::path_join(SERIALIZED_ZOO, "onnx/provenance_tag_add.prototxt")); file_util::path_join(SERIALIZED_ZOO, "onnx/provenance_tag_add.prototxt"));
const auto ng_nodes = function->get_ordered_ops(); const auto ng_nodes = function->get_ordered_ops();
for (const auto ng_node : ng_nodes) for (auto ng_node : ng_nodes)
{ {
for (const auto tag : ng_node->get_provenance_tags()) for (auto tag : ng_node->get_provenance_tags())
{ {
EXPECT_HAS_SUBSTRING(tag, "ONNX"); EXPECT_HAS_SUBSTRING(tag, "ONNX");
} }
...@@ -52,7 +52,7 @@ void test_provenance_tags(const std::shared_ptr<Function> function, ...@@ -52,7 +52,7 @@ void test_provenance_tags(const std::shared_ptr<Function> function,
const std::string& expected_provenance_tag) const std::string& expected_provenance_tag)
{ {
int node_count = 0; int node_count = 0;
for (const auto ng_node : function->get_ordered_ops()) for (auto ng_node : function->get_ordered_ops())
{ {
if (as_type_ptr<NodeToCheck>(ng_node)) if (as_type_ptr<NodeToCheck>(ng_node))
{ {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment