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
5d9bb084
Commit
5d9bb084
authored
Sep 17, 2018
by
Paul
Browse files
Tidy fixes
parent
8068c5a6
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
src/onnx/onnx.cpp
src/onnx/onnx.cpp
+1
-1
No files found.
src/onnx/onnx.cpp
View file @
5d9bb084
...
@@ -102,7 +102,7 @@ struct onnx_parser
...
@@ -102,7 +102,7 @@ struct onnx_parser
}
}
instruction_ref
instruction_ref
parse_softmax
(
const
std
::
string
&
,
attribute_map
,
std
::
vector
<
instruction_ref
>
args
)
parse_softmax
(
const
std
::
string
&
,
const
attribute_map
&
,
std
::
vector
<
instruction_ref
>
args
)
{
{
auto
dims
=
args
.
front
()
->
get_shape
().
lens
();
auto
dims
=
args
.
front
()
->
get_shape
().
lens
();
auto
r
=
prog
.
add_instruction
(
reshape
{{
long
(
dims
[
0
]),
1
,
1
,
long
(
dims
[
1
])}},
args
.
front
());
auto
r
=
prog
.
add_instruction
(
reshape
{{
long
(
dims
[
0
]),
1
,
1
,
long
(
dims
[
1
])}},
args
.
front
());
...
...
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