Initial

0317936f · charlie · 8343d47d · 0317936f
Commit 0317936f authored Aug 10, 2022 by charlie
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 0 deletions

src/include/migraphx/op/nonzero.hpp src/include/migraphx/op/nonzero.hpp +8 -0

No files found.
--- a/src/include/migraphx/op/nonzero.hpp
+++ b/src/include/migraphx/op/nonzero.hpp
@@ -39,6 +39,14 @@ namespace op {
 struct nonzero
 {
+    bool use_dynamic = false;
+    template <class Self, class F>
+    static auto reflect(Self& self, F f)
+    {
+        return pack(f(self.use_dynamic, "use_dynamic"));
+    }
    std::string name() const { return "nonzero"; }
    shape compute_shape(std::vector<shape> inputs) const