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
45752b0f
Commit
45752b0f
authored
6 years ago
by
Nick Korovaiko
Committed by
Scott Cyphers
6 years ago
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
switch more passes to static_pointer_cast (#2041)
parent
cccdc304
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
4 deletions
+4
-4
cpu_rnn_fusion.cpp
src/ngraph/runtime/cpu/pass/cpu_rnn_fusion.cpp
+1
-1
cpu_workspace_insertion.cpp
src/ngraph/runtime/cpu/pass/cpu_workspace_insertion.cpp
+3
-3
No files found.
src/ngraph/runtime/cpu/pass/cpu_rnn_fusion.cpp
View file @
45752b0f
...
...
@@ -530,7 +530,7 @@ void ngraph::runtime::cpu::pass::RNNFusion::construct_rnn_lstm_fprop()
// now get the GOE0 which is the first output of lstm (ht)
for
(
auto
&
goes
:
lstm_nodes
[
index
]
->
get_outputs
().
at
(
0
).
get_inputs
())
{
auto
goe_node
=
std
::
dynam
ic_pointer_cast
<
op
::
GetOutputElement
>
(
goes
->
get_node
());
auto
goe_node
=
std
::
stat
ic_pointer_cast
<
op
::
GetOutputElement
>
(
goes
->
get_node
());
// first output node of lstm
if
(
goe_node
->
get_n
()
==
0
)
{
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/runtime/cpu/pass/cpu_workspace_insertion.cpp
View file @
45752b0f
...
...
@@ -94,13 +94,13 @@ bool runtime::cpu::pass::CPUWorkspaceInsertion::run_on_function(std::shared_ptr<
bool
runtime
::
cpu
::
pass
::
CPUWorkspaceInsertion
::
transform
(
pattern
::
Matcher
&
m
)
{
auto
data
=
std
::
dynam
ic_pointer_cast
<
pattern
::
op
::
Label
>
(
m
.
get_pattern
()
->
get_argument
(
0
));
auto
delta
=
std
::
dynam
ic_pointer_cast
<
pattern
::
op
::
Label
>
(
m
.
get_pattern
()
->
get_argument
(
1
));
auto
data
=
std
::
stat
ic_pointer_cast
<
pattern
::
op
::
Label
>
(
m
.
get_pattern
()
->
get_argument
(
0
));
auto
delta
=
std
::
stat
ic_pointer_cast
<
pattern
::
op
::
Label
>
(
m
.
get_pattern
()
->
get_argument
(
1
));
NGRAPH_DEBUG
<<
"In a callback for construct_max_pool_with_indices against "
<<
m
.
get_match_root
()
->
get_name
();
auto
pattern_map
=
m
.
get_pattern_map
();
auto
m_max_pool_bprop
=
std
::
dynam
ic_pointer_cast
<
op
::
MaxPoolBackprop
>
(
m
.
get_match_root
());
auto
m_max_pool_bprop
=
std
::
stat
ic_pointer_cast
<
op
::
MaxPoolBackprop
>
(
m
.
get_match_root
());
if
(
m_max_pool_bprop
->
get_shape
().
size
()
!=
4
||
m_max_pool_bprop
->
get_window_shape
().
size
()
!=
2
||
...
...
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