[ONNX] Add valid padding support (#2628)

665b9c95 · tsocha · Scott Cyphers · 80b4b1da · 665b9c95
Commit 665b9c95 authored Mar 18, 2019 by tsocha Committed by Scott Cyphers Mar 18, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

convpool.cpp src/ngraph/frontend/onnx_import/utils/convpool.cpp +4 -0

No files found.
--- a/src/ngraph/frontend/onnx_import/utils/convpool.cpp
+++ b/src/ngraph/frontend/onnx_import/utils/convpool.cpp
@@ -93,6 +93,10 @@ namespace ngraph
                                             const Strides& strides,
                                             const std::string& auto_pad)
                {
+                    if (auto_pad == "VALID")
+                    {
+                        return CoordinateDiff(input_shape.size());
+                    }
                    CoordinateDiff pads_begin;
                    CoordinateDiff pads_end;
                    // Omit {N,C} axes