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
8b023ae3
Commit
8b023ae3
authored
Dec 11, 2018
by
Khalique
Browse files
fix beta val
parent
ef0fe6e6
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
5 deletions
+9
-5
src/onnx/onnx.cpp
src/onnx/onnx.cpp
+9
-5
No files found.
src/onnx/onnx.cpp
View file @
8b023ae3
...
...
@@ -400,7 +400,7 @@ struct onnx_parser
parse_gemm
(
const
std
::
string
&
,
attribute_map
attributes
,
std
::
vector
<
instruction_ref
>
args
)
{
float
alpha
=
1.0
f
;
float
beta
=
0
.0
f
;
float
beta
=
1
.0
f
;
bool
transa
=
false
;
bool
transb
=
false
;
if
(
contains
(
attributes
,
"alpha"
))
...
...
@@ -427,10 +427,14 @@ struct onnx_parser
if
(
beta
!=
0.
f
)
{
auto
l3
=
prog
.
add_instruction
(
op
::
dot
{
alpha
},
l1
,
l2
);
auto
beta_val
=
prog
.
add_literal
(
beta
);
auto
l4
=
prog
.
add_instruction
(
op
::
scalar
{
args
[
2
]
->
get_shape
()},
beta_val
);
auto
l5
=
prog
.
add_instruction
(
op
::
mul
{},
args
[
2
],
l4
);
return
add_broadcastable_binary_op
(
l3
,
l5
,
op
::
add
{});
auto
l4
=
args
[
2
];
if
(
beta
==
1.
f
)
{
auto
beta_val
=
prog
.
add_literal
(
beta
);
auto
l5
=
prog
.
add_instruction
(
op
::
scalar
{
args
[
2
]
->
get_shape
()},
beta_val
);
l4
=
prog
.
add_instruction
(
op
::
mul
{},
args
[
2
],
l5
);
}
return
add_broadcastable_binary_op
(
l3
,
l4
,
op
::
add
{});
}
}
return
prog
.
add_instruction
(
op
::
dot
{
alpha
,
beta
},
l1
,
l2
);
...
...
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