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)
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 val = make_shared<pattern::op::Label>(iconst0);
......@@ -53,8 +53,7 @@ void pass::CoreFusion::construct_relu_pattern()
auto max = make_shared<op::Maximum>(skip_broadcast, val);
pattern::graph_rewrite_callback callback = [val, zero](pattern::Matcher& m) {
NGRAPH_DEBUG << "In a callback for construct_relu_pattern against "
<< m.match_root()->get_name();
NGRAPH_DEBUG << "In a callback for construct_relu against " << m.match_root()->get_name();
auto pattern_map = m.get_pattern_map();
auto mzero = m.get_pattern_map()[zero];
......
......@@ -32,7 +32,7 @@ public:
CoreFusion()
: 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