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
ca4eb13d
Commit
ca4eb13d
authored
Jan 31, 2019
by
Khalique
Browse files
formatting
parent
5cd9877d
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
7 deletions
+6
-7
src/tf/tf.cpp
src/tf/tf.cpp
+6
-7
No files found.
src/tf/tf.cpp
View file @
ca4eb13d
...
...
@@ -98,7 +98,7 @@ struct tf_parser
{
if
(
is_nhwc
)
{
l0
=
prog
.
add_instruction
(
op
::
transpose
{{
0
,
3
,
1
,
2
}},
args
[
1
]);
l0
=
prog
.
add_instruction
(
op
::
transpose
{{
0
,
3
,
1
,
2
}},
args
[
1
]);
}
}
return
add_broadcastable_binary_op
(
args
[
0
],
l0
,
x
);
...
...
@@ -162,8 +162,7 @@ struct tf_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
;
op
::
batch_norm_inference
::
bn_infer_mode_t
bn_mode
=
op
::
batch_norm_inference
::
spatial
;
if
(
contains
(
attributes
,
"epsilon"
))
{
epsilon
=
attributes
.
at
(
"epsilon"
).
f
();
...
...
@@ -385,7 +384,7 @@ struct tf_parser
auto
l0
=
args
[
0
];
if
(
is_nhwc
)
{
l0
=
prog
.
add_instruction
(
op
::
transpose
{{
0
,
2
,
3
,
1
}},
args
[
0
]);
l0
=
prog
.
add_instruction
(
op
::
transpose
{{
0
,
2
,
3
,
1
}},
args
[
0
]);
}
return
prog
.
add_instruction
(
op
,
l0
);
}
...
...
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