Skip to content

Commit

Permalink
Update src/frontends/pytorch/src/utils.cpp
Browse files Browse the repository at this point in the history
  • Loading branch information
mvafin authored Jan 17, 2025
1 parent e04dbde commit f3bd20c
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/frontends/pytorch/src/utils.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -56,6 +56,8 @@ void num_inputs_check(const NodeContext& context, size_t min_inputs, size_t max_
PYTORCH_OP_CONVERSION_CHECK(!complex_type_mark, "The operation doesn't allow complex type.");
}
}

// Check that additional inputs are all None, otherwise raise exception
for (auto i = max_inputs; i < num_inputs; i++) {
FRONT_END_OP_CONVERSION_CHECK(context.input_is_none(i), "Got more inputs than expected: ", i + 1);
}
Expand Down

0 comments on commit f3bd20c

Please sign in to comment.