Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
MIGraphX
Commits
82b0a072
Commit
82b0a072
authored
Nov 08, 2018
by
Khalique
Browse files
added dropout onnx file
parent
70a993f3
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
12 additions
and
0 deletions
+12
-0
test/onnx/onnx_test.cpp
test/onnx/onnx_test.cpp
+12
-0
No files found.
test/onnx/onnx_test.cpp
View file @
82b0a072
...
...
@@ -158,6 +158,17 @@ void transpose_test()
EXPECT
(
p
==
prog
);
}
void
dropout_test
()
{
migraph
::
program
p
;
auto
input
=
p
.
add_parameter
(
"0"
,
migraph
::
shape
{
migraph
::
shape
::
float_type
,
{
1
,
3
,
2
,
2
}});
p
.
add_instruction
(
migraph
::
op
::
identity
{},
input
);
auto
prog
=
migraph
::
parse_onnx
(
"dropout_test.onnx"
);
EXPECT
(
p
==
prog
);
}
int
main
()
{
pytorch_conv_bias_test
();
...
...
@@ -169,4 +180,5 @@ int main()
globalavgpool_test
();
globalmaxpool_test
();
transpose_test
();
dropout_test
();
}
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