File tree 1 file changed +14
-0
lines changed
1 file changed +14
-0
lines changed Original file line number Diff line number Diff line change @@ -81,3 +81,17 @@ TEST(type_prop, swish_2_inputs)
81
81
ASSERT_TRUE (swish_func->get_output_partial_shape (0 ).same_scheme (data->get_output_shape (0 )));
82
82
ASSERT_TRUE (swish_func->get_output_partial_shape (0 ).rank ().is_static ());
83
83
}
84
+
85
+ TEST (type_prop, swish_incompatible_type_boolean)
86
+ {
87
+ auto data = make_shared<op::Parameter>(element::boolean, Shape{1 , 3 , 6 });
88
+ auto beta = make_shared<op::Parameter>(element::f32, Shape{});
89
+ ASSERT_THROW (make_shared<op::v4::Swish>(data, beta);, ngraph::NodeValidationFailure);
90
+ }
91
+
92
+ TEST (type_prop, swish_incompatible_types_u32)
93
+ {
94
+ auto data = make_shared<op::Parameter>(element::f32, Shape{1 , 3 , 6 });
95
+ auto beta = make_shared<op::Parameter>(element::u32, Shape{});
96
+ ASSERT_THROW (make_shared<op::v4::Swish>(data, beta);, ngraph::NodeValidationFailure);
97
+ }
You can’t perform that action at this time.
0 commit comments