Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
gaoqiong
MIGraphX
Commits
e1876299
"dockerfile/rocm5.0-pytorch1.9.0.dockerfile" did not exist on "ff563b66af8a4fa2e99a25d994a9a04a97388742"
Commit
e1876299
authored
Jan 29, 2019
by
Shucai Xiao
Browse files
add cpu test for the rnn operator.
parent
e551275d
Changes
2
Expand all
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
272 additions
and
82 deletions
+272
-82
src/onnx/onnx.cpp
src/onnx/onnx.cpp
+1
-1
test/cpu_ops_test.cpp
test/cpu_ops_test.cpp
+271
-81
No files found.
src/onnx/onnx.cpp
View file @
e1876299
...
...
@@ -798,7 +798,7 @@ struct onnx_parser
// For RNN, LSTM, and GRU operators, one of the input arguments
// is prim::Undefined, and it is ignored by protobuf. We use a
// hack to ignore this argument for these three operators
std
::
string
op_type
=
node
.
op_type
();
const
std
::
string
op_type
=
node
.
op_type
();
if
((
op_type
==
"RNN"
||
op_type
==
"LSTM"
||
op_type
==
"GRU"
)
&&
input
.
empty
()
==
true
)
{
...
...
test/cpu_ops_test.cpp
View file @
e1876299
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