IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add invalid input data check

Browse Source
This commit is contained in:
DD 2021-06-01 10:36:16 +08:00
parent e866ba7b76
commit 44de64628e
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ngraph/core/src/op/softplus.cpp
View File

@ -29,6 +29,11 @@ bool op::v4::SoftPlus::visit_attributes(AttributeVisitor& visitor)
void op::v4::SoftPlus::validate_and_infer_types() void op::v4::SoftPlus::validate_and_infer_types()
{ {
NGRAPH_OP_SCOPE(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 element type must be float. Got: ",
input_et);
set_output_size(1); set_output_size(1);
set_output_type(0, get_input_element_type(0), get_input_partial_shape(0)); set_output_type(0, get_input_element_type(0), get_input_partial_shape(0));
} }