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
80db037c
Commit
80db037c
authored
Oct 09, 2018
by
wsttiger
Browse files
Formatting
parent
ad079eb8
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
13 additions
and
11 deletions
+13
-11
test/fwd_conv_batchnorm_rewrite_test.cpp
test/fwd_conv_batchnorm_rewrite_test.cpp
+13
-11
No files found.
test/fwd_conv_batchnorm_rewrite_test.cpp
View file @
80db037c
...
@@ -34,17 +34,19 @@ void fwd_conv_batchnorm_rewrite_test()
...
@@ -34,17 +34,19 @@ void fwd_conv_batchnorm_rewrite_test()
migraph
::
shape
ws
{
migraph
::
shape
::
float_type
,
{
1
,
3
,
3
,
3
}};
migraph
::
shape
ws
{
migraph
::
shape
::
float_type
,
{
1
,
3
,
3
,
3
}};
migraph
::
shape
vars
{
migraph
::
shape
::
float_type
,
{
1
}};
migraph
::
shape
vars
{
migraph
::
shape
::
float_type
,
{
1
}};
auto
create_program
=
[
&
]
()
{
auto
create_program
=
migraph
::
program
p
;
[
&
]()
{
auto
x
=
p
.
add_literal
(
xs
,
xdata
);
migraph
::
program
p
;
auto
w
=
p
.
add_literal
(
ws
,
wdata
);
auto
x
=
p
.
add_literal
(
xs
,
xdata
);
auto
conv
=
p
.
add_instruction
(
migraph
::
op
::
convolution
{{
0
,
0
},
{
1
,
1
},
{
1
,
1
}},
x
,
w
);
auto
w
=
p
.
add_literal
(
ws
,
wdata
);
auto
scale
=
p
.
add_literal
(
migraph
::
literal
{
vars
,
{
3.0
f
}});
auto
conv
=
p
.
add_instruction
(
migraph
::
op
::
convolution
{{
0
,
0
},
{
1
,
1
},
{
1
,
1
}},
x
,
w
);
auto
bias
=
p
.
add_literal
(
migraph
::
literal
{
vars
,
{
8.1
f
}});
auto
scale
=
p
.
add_literal
(
migraph
::
literal
{
vars
,
{
3.0
f
}});
auto
mean
=
p
.
add_literal
(
migraph
::
literal
{
vars
,
{
4.0
f
}});
auto
bias
=
p
.
add_literal
(
migraph
::
literal
{
vars
,
{
8.1
f
}});
auto
variance
=
p
.
add_literal
(
migraph
::
literal
{
vars
,
{
37.11
f
}});
auto
mean
=
p
.
add_literal
(
migraph
::
literal
{
vars
,
{
4.0
f
}});
p
.
add_instruction
(
migraph
::
op
::
batch_norm_inference
{},
conv
,
scale
,
bias
,
mean
,
variance
);
auto
variance
=
p
.
add_literal
(
migraph
::
literal
{
vars
,
{
37.11
f
}});
}
p
.
add_instruction
(
migraph
::
op
::
batch_norm_inference
{},
conv
,
scale
,
bias
,
mean
,
variance
);
}
migraph
::
program
p1
=
create_program
();
migraph
::
program
p1
=
create_program
();
migraph
::
program
p2
=
create_program
();
migraph
::
program
p2
=
create_program
();
...
...
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