Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in / Register
Toggle navigation
N
ngraph
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Packages
Packages
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
submodule
ngraph
Commits
5648af4e
Commit
5648af4e
authored
6 years ago
by
Nick Korovaiko
Committed by
Robert Kimball
6 years ago
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
construct_relu_pattern -> construct_relu (#932)
parent
664969f5
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
5 deletions
+4
-5
core_fusion.cpp
src/ngraph/pass/core_fusion.cpp
+2
-3
core_fusion.hpp
src/ngraph/pass/core_fusion.hpp
+2
-2
No files found.
src/ngraph/pass/core_fusion.cpp
View file @
5648af4e
...
...
@@ -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
];
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/pass/core_fusion.hpp
View file @
5648af4e
...
...
@@ -32,7 +32,7 @@ public:
CoreFusion
()
:
GraphRewrite
()
{
construct_relu
_pattern
();
construct_relu
();
}
void
construct_relu
_pattern
();
void
construct_relu
();
};
This diff is collapsed.
Click to expand it.
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment