changed padding mode to enum

210c9839 · Khalique · aaf0f9e1 · 210c9839
Commit 210c9839 authored Jan 24, 2019 by Khalique
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 1 deletion

src/include/migraphx/operators.hpp src/include/migraphx/operators.hpp +7 -1

No files found.
--- a/src/include/migraphx/operators.hpp
+++ b/src/include/migraphx/operators.hpp
@@ -614,7 +614,13 @@ struct pad
 {
    std::vector<int64_t> pads;
    float value      = 0.0f;
-    std::string mode = "constant";
+    enum padding_mode_t
+    {
+        constant_, //NOLINT
+        reflect_, //NOLINT
+        edge
+    };
+    padding_mode_t mode = constant_;

    template <class Self, class F>
    static auto reflect(Self& self, F f)