CPU Direct Execution: Migrate AvgPool to auto-registration

9376bd0c · Jaikrishnan Menon · aebd416a · 9376bd0c
Commit 9376bd0c authored Jul 20, 2018 by Jaikrishnan Menon
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

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

No files found.
--- a/src/ngraph/runtime/cpu/builder/avg_pool.cpp
+++ b/src/ngraph/runtime/cpu/builder/avg_pool.cpp
@@ -107,6 +107,8 @@ namespace ngraph
                    functors.emplace_back(functor);
                }
            }
+            REGISTER_OP_BUILDER(AvgPool);
        }
    }
 }