Commit ff9c397c authored by charlie's avatar charlie
Browse files

check_shapes allow dynamic

parent abc625e7
...@@ -58,7 +58,7 @@ struct nonmaxsuppression ...@@ -58,7 +58,7 @@ struct nonmaxsuppression
shape compute_shape(std::vector<shape> inputs) const shape compute_shape(std::vector<shape> inputs) const
{ {
// requires at least 2 inputs // requires at least 2 inputs
check_shapes{{inputs.at(0), inputs.at(1)}, *this}.only_dims(3).same_ndims(); check_shapes{{inputs.at(0), inputs.at(1)}, *this, true}.only_dims(3).same_ndims();
const auto max_num_boxes = inputs.at(0).max_lens().at(1); const auto max_num_boxes = inputs.at(0).max_lens().at(1);
auto fixed_shape_error_check = [&]() { auto fixed_shape_error_check = [&]() {
......
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