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
1a2e5c11
Commit
1a2e5c11
authored
Dec 05, 2018
by
Khalique
Browse files
formatting
parent
13c9ed9c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
2 deletions
+3
-2
test/onnx/onnx_test.cpp
test/onnx/onnx_test.cpp
+3
-2
No files found.
test/onnx/onnx_test.cpp
View file @
1a2e5c11
...
...
@@ -382,7 +382,8 @@ void squeeze_unsqueeze_test()
migraphx
::
program
p
;
std
::
vector
<
int64_t
>
squeeze_axes
{
0
,
2
,
3
,
5
};
std
::
vector
<
int64_t
>
unsqueeze_axes
{
0
,
1
,
3
,
5
};
auto
l0
=
p
.
add_parameter
(
"0"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
1
,
3
,
1
,
1
,
2
,
1
}});
auto
l0
=
p
.
add_parameter
(
"0"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
1
,
3
,
1
,
1
,
2
,
1
}});
auto
l1
=
p
.
add_instruction
(
migraphx
::
op
::
squeeze
{
squeeze_axes
},
l0
);
p
.
add_instruction
(
migraphx
::
op
::
unsqueeze
{
unsqueeze_axes
},
l1
);
auto
prog
=
migraphx
::
parse_onnx
(
"squeeze_unsqueeze_test.onnx"
);
...
...
@@ -405,7 +406,7 @@ void slice_test()
{
migraphx
::
program
p
;
auto
l0
=
p
.
add_parameter
(
"0"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
3
,
2
}});
p
.
add_instruction
(
migraphx
::
op
::
slice
{{
0
,
1
},
{
1
,
0
},
{
2
,
2
}},
l0
);
p
.
add_instruction
(
migraphx
::
op
::
slice
{{
0
,
1
},
{
1
,
0
},
{
2
,
2
}},
l0
);
auto
prog
=
migraphx
::
parse_onnx
(
"slice_test.onnx"
);
EXPECT
(
p
==
prog
);
...
...
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