Commit 5648af4e authored by Nick Korovaiko's avatar Nick Korovaiko Committed by Robert Kimball

construct_relu_pattern -> construct_relu (#932)

parent 664969f5
...@@ -40,7 +40,7 @@ static shared_ptr<Node> construct_constant_node(int n) ...@@ -40,7 +40,7 @@ static shared_ptr<Node> construct_constant_node(int n)
return op::Constant::create(element::f32, Shape{}, {n}); return op::Constant::create(element::f32, Shape{}, {n});
} }
void pass::CoreFusion::construct_relu_pattern() void pass::CoreFusion::construct_relu()
{ {
auto iconst0 = construct_constant_node(0); auto iconst0 = construct_constant_node(0);
auto val = make_shared<pattern::op::Label>(iconst0); auto val = make_shared<pattern::op::Label>(iconst0);
...@@ -53,8 +53,7 @@ void pass::CoreFusion::construct_relu_pattern() ...@@ -53,8 +53,7 @@ void pass::CoreFusion::construct_relu_pattern()
auto max = make_shared<op::Maximum>(skip_broadcast, val); auto max = make_shared<op::Maximum>(skip_broadcast, val);
pattern::graph_rewrite_callback callback = [val, zero](pattern::Matcher& m) { pattern::graph_rewrite_callback callback = [val, zero](pattern::Matcher& m) {
NGRAPH_DEBUG << "In a callback for construct_relu_pattern against " NGRAPH_DEBUG << "In a callback for construct_relu against " << m.match_root()->get_name();
<< m.match_root()->get_name();
auto pattern_map = m.get_pattern_map(); auto pattern_map = m.get_pattern_map();
auto mzero = m.get_pattern_map()[zero]; auto mzero = m.get_pattern_map()[zero];
......
...@@ -32,7 +32,7 @@ public: ...@@ -32,7 +32,7 @@ public:
CoreFusion() CoreFusion()
: GraphRewrite() : GraphRewrite()
{ {
construct_relu_pattern(); construct_relu();
} }
void construct_relu_pattern(); void construct_relu();
}; };
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment