Fix random segfault when LRN cpu fallback to reference kernel (#3986)

bc23562a · Robert Kimball · Scott Cyphers · 4a1e308a · bc23562a
Commit bc23562a authored Dec 04, 2019 by Robert Kimball Committed by Scott Cyphers Dec 04, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

lrn.cpp src/ngraph/runtime/cpu/builder/lrn.cpp +2 -0

No files found.
--- a/src/ngraph/runtime/cpu/builder/lrn.cpp
+++ b/src/ngraph/runtime/cpu/builder/lrn.cpp
@@ -93,6 +93,7 @@ namespace ngraph
                                   beta,
                                   bias,
                                   arg_shape,
+                                   axes,
                                   axes_shape,
                                   nsize,
                                   arg_buffer_index,
@@ -116,6 +117,7 @@ namespace ngraph
                                   beta,
                                   bias,
                                   arg_shape,
+                                   axes,
                                   axes_shape,
                                   nsize,
                                   arg_buffer_index,