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
b05e573d
"vscode:/vscode.git/clone" did not exist on "41e976edde8920db7db82217e920ab50c465b6ee"
Commit
b05e573d
authored
Aug 23, 2019
by
Khalique
Browse files
remove is_test
parent
d1537169
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
6 deletions
+0
-6
src/onnx/onnx.cpp
src/onnx/onnx.cpp
+0
-6
No files found.
src/onnx/onnx.cpp
View file @
b05e573d
...
...
@@ -655,7 +655,6 @@ struct onnx_parser
float
epsilon
=
1e-5
f
;
float
momentum
=
0.9
f
;
op
::
batch_norm_inference
::
bn_infer_mode_t
bn_mode
=
op
::
batch_norm_inference
::
spatial
;
bool
is_test
=
false
;
if
(
contains
(
attributes
,
"epsilon"
))
{
epsilon
=
parse_value
(
attributes
.
at
(
"epsilon"
)).
at
<
float
>
();
...
...
@@ -664,17 +663,12 @@ struct onnx_parser
{
momentum
=
parse_value
(
attributes
.
at
(
"momentum"
)).
at
<
float
>
();
}
if
(
contains
(
attributes
,
"is_test"
))
{
is_test
=
parse_value
(
attributes
.
at
(
"is_test"
)).
at
<
uint64_t
>
()
>
0
;
}
if
(
contains
(
attributes
,
"spatial"
))
{
bn_mode
=
(
parse_value
(
attributes
.
at
(
"spatial"
)).
at
<
uint64_t
>
()
>
0
)
?
op
::
batch_norm_inference
::
spatial
:
op
::
batch_norm_inference
::
per_activation
;
}
(
void
)
is_test
;
op
::
batch_norm_inference
op
{
epsilon
,
momentum
,
bn_mode
};
return
prog
.
add_instruction
(
op
,
std
::
move
(
args
));
}
...
...
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