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
23d0deb0
Commit
23d0deb0
authored
Mar 04, 2019
by
Shucai Xiao
Browse files
Merge branch 'develop' of
https://github.com/ROCmSoftwarePlatform/AMDMIGraphX
into seq2seq_example
parents
15dda77e
8274cb47
Changes
5
Hide whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
15 additions
and
15 deletions
+15
-15
src/onnx/onnx.cpp
src/onnx/onnx.cpp
+2
-2
test/onnx/constant_scalar.onnx
test/onnx/constant_scalar.onnx
+4
-4
test/onnx/onnx_test.cpp
test/onnx/onnx_test.cpp
+2
-2
test/onnx/sum_test.onnx
test/onnx/sum_test.onnx
+3
-3
test/onnx/unknown_test.onnx
test/onnx/unknown_test.onnx
+4
-4
No files found.
src/onnx/onnx.cpp
View file @
23d0deb0
...
@@ -1164,9 +1164,9 @@ struct onnx_parser
...
@@ -1164,9 +1164,9 @@ struct onnx_parser
instructions
[
name
]
=
prog
.
add_parameter
(
name
,
s
);
instructions
[
name
]
=
prog
.
add_parameter
(
name
,
s
);
}
}
}
}
for
(
auto
&&
p
:
nodes
)
for
(
auto
&&
output
:
graph
.
output
()
)
{
{
this
->
parse_node
(
p
.
first
);
this
->
parse_node
(
output
.
name
()
);
}
}
}
}
...
...
test/onnx/constant_scalar.onnx
View file @
23d0deb0
shape-gathe
r-example:
O
constant-scala
r-example:
R
2value
"Constant*
00
"Constant*
!
value*
*Bconst_tensor
constantb
value*
*Bconst_tensor
test-
constantb
z
0
B
B
\ No newline at end of file
test/onnx/onnx_test.cpp
View file @
23d0deb0
...
@@ -470,8 +470,8 @@ TEST_CASE(flatten_test)
...
@@ -470,8 +470,8 @@ TEST_CASE(flatten_test)
{
{
migraphx
::
program
p
;
migraphx
::
program
p
;
auto
l0
=
p
.
add_parameter
(
"0"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
2
,
3
,
4
,
5
}});
auto
l0
=
p
.
add_parameter
(
"0"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
2
,
3
,
4
,
5
}});
p
.
add_instruction
(
migraphx
::
op
::
flatten
{
1
},
l0
);
p
.
add_instruction
(
migraphx
::
op
::
flatten
{
2
},
l0
);
p
.
add_instruction
(
migraphx
::
op
::
flatten
{
2
},
l0
);
p
.
add_instruction
(
migraphx
::
op
::
flatten
{
1
},
l0
);
auto
prog
=
migraphx
::
parse_onnx
(
"flatten_test.onnx"
);
auto
prog
=
migraphx
::
parse_onnx
(
"flatten_test.onnx"
);
EXPECT
(
p
==
prog
);
EXPECT
(
p
==
prog
);
...
@@ -524,7 +524,7 @@ TEST_CASE(constant_test)
...
@@ -524,7 +524,7 @@ TEST_CASE(constant_test)
TEST_CASE
(
constant_test_scalar
)
TEST_CASE
(
constant_test_scalar
)
{
{
migraphx
::
program
p
;
migraphx
::
program
p
;
p
.
add_literal
(
migraphx
::
literal
{
migraphx
::
shape
{
migraphx
::
shape
::
int32_type
},
{
1
}});
p
.
add_literal
(
migraphx
::
literal
{
migraphx
::
shape
{
migraphx
::
shape
::
int32_type
,
{
1
}
},
{
1
}});
auto
prog
=
migraphx
::
parse_onnx
(
"constant_scalar.onnx"
);
auto
prog
=
migraphx
::
parse_onnx
(
"constant_scalar.onnx"
);
EXPECT
(
p
==
prog
);
EXPECT
(
p
==
prog
);
...
...
test/onnx/sum_test.onnx
View file @
23d0deb0
sum-example:
e
sum-example:
a
0
0
1
1
23"Sum
test-
dropout
Z
23"Sum
test-
sum
Z
0
0
...
@@ -15,7 +15,7 @@
...
@@ -15,7 +15,7 @@
b
b
2
3
B
B
\ No newline at end of file
test/onnx/unknown_test.onnx
View file @
23d0deb0
unknown-example:
unknown-example:
0
0
12"Unknown
12"Unknown
2"Unknowntest-unknownZ
2
3
"Unknowntest-unknownZ
0
0
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
b
b
2
3
...
...
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