Allow nonstandard shapes

ba536652 · charlie · c56ed4e9 · ba536652
Commit ba536652 authored Jun 07, 2022 by charlie
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/include/migraphx/op/nonmaxsuppression.hpp src/include/migraphx/op/nonmaxsuppression.hpp +1 -1

No files found.
--- a/src/include/migraphx/op/nonmaxsuppression.hpp
+++ b/src/include/migraphx/op/nonmaxsuppression.hpp
@@ -33,7 +33,7 @@ struct nonmaxsuppression
    shape compute_shape(std::vector<shape> inputs) const
    {
        // requires at least 2 inputs
-        check_shapes{inputs, *this}.standard();
+        check_shapes{inputs, *this};
        check_shapes{{inputs.at(0), inputs.at(1)}, *this}.only_dims(3);
        auto lens = inputs.front().lens();