fix lrn for dynshape (#4057)

438f7f5f · baojun · Scott Cyphers · 796a5a61 · 438f7f5f
Commit 438f7f5f authored Dec 16, 2019 by baojun Committed by Scott Cyphers Dec 16, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

lrn.cpp src/ngraph/op/lrn.cpp +2 -1

No files found.
--- a/src/ngraph/op/lrn.cpp
+++ b/src/ngraph/op/lrn.cpp
@@ -79,7 +79,8 @@ void op::LRN::validate_and_infer_types()
    NODE_VALIDATION_CHECK(
        this,
-        static_cast<size_t>(axes_shape[0]) <= static_cast<size_t>(input_shape_rank),
+        axes_shape.is_dynamic() || input_shape_rank.is_dynamic() ||
+            static_cast<size_t>(axes_shape[0]) <= static_cast<size_t>(input_shape_rank),
        "Number of elements of axes must be >= 0 and <= argument rank (axes_shape[0]: ",
        axes_shape[0],
        ").");