add input type dynamic check

2021-06-01 11:15:23 +08:00 · 2021-06-01 11:15:23 +08:00 · 5168ad53e3
commit 5168ad53e3
parent 44de64628e
1 changed files with 3 additions and 1 deletions
--- a/ngraph/core/src/op/softplus.cpp
+++ b/ngraph/core/src/op/softplus.cpp
@ -30,10 +30,12 @@ void op::v4::SoftPlus::validate_and_infer_types()
 {
    NGRAPH_OP_SCOPE(v4_SoftPlus_validate_and_infer_types);
    const element::Type& input_et = get_input_element_type(0);
+
    NODE_VALIDATION_CHECK(this,
-                          input_et.is_real(),
+                          input_et.is_dynamic() || input_et.is_real(),
                          "Input element type must be float. Got: ",
                          input_et);
+
    set_output_size(1);
    set_output_type(0, get_input_element_type(0), get_input_partial_shape(0));
 }