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
df2da5de
Commit
df2da5de
authored
Nov 05, 2021
by
Shucai Xiao
Browse files
additional refinement of print out info
parent
6323fc2c
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
tools/test_runner.py
tools/test_runner.py
+1
-1
No files found.
tools/test_runner.py
View file @
df2da5de
...
@@ -221,7 +221,6 @@ def main():
...
@@ -221,7 +221,6 @@ def main():
# get param names
# get param names
param_names
=
model_parameter_names
(
model_path_name
)
param_names
=
model_parameter_names
(
model_path_name
)
print
(
"param_name = {}"
.
format
(
param_names
))
# get output names
# get output names
output_names
=
model_output_names
(
model_path_name
)
output_names
=
model_output_names
(
model_path_name
)
...
@@ -232,6 +231,7 @@ def main():
...
@@ -232,6 +231,7 @@ def main():
param_shapes
=
get_input_shapes
(
sample_case
,
param_names
)
param_shapes
=
get_input_shapes
(
sample_case
,
param_names
)
for
name
,
dims
in
param_shapes
.
items
():
for
name
,
dims
in
param_shapes
.
items
():
print
(
"Input: {}, shape: {}"
.
format
(
name
,
dims
))
print
(
"Input: {}, shape: {}"
.
format
(
name
,
dims
))
print
()
# read and compile model
# read and compile model
model
=
migraphx
.
parse_onnx
(
model_path_name
,
map_input_dims
=
param_shapes
)
model
=
migraphx
.
parse_onnx
(
model_path_name
,
map_input_dims
=
param_shapes
)
...
...
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