Skip to content

GitLab

Commit 3c100748 authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

formatting

parent 01753a1b
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 18 deletions
test/layout_nhwc.cpp
View file @ 3c100748
......@@ -129,16 +129,10 @@ TEST_CASE(conv_conv)
migraphx::module m1;
{
auto x = m1.add_parameter("x", {migraphx::shape::float_type, {1, 8, 16, 16}});
auto w = m1.add_literal(
migraphx::generate_literal({migraphx::shape::float_type, {8, 8, 1, 1}}));
auto conv1 = m1.add_instruction(
migraphx::make_op("convolution"),
x,
w);
m1.add_instruction(
migraphx::make_op("convolution"),
conv1,
w);
auto w =
m1.add_literal(migraphx::generate_literal({migraphx::shape::float_type, {8, 8, 1, 1}}));
auto conv1 = m1.add_instruction(migraphx::make_op("convolution"), x, w);
m1.add_instruction(migraphx::make_op("convolution"), conv1, w);
}
run_pass(m1);
......@@ -146,14 +140,11 @@ TEST_CASE(conv_conv)
{
auto x = add_layout_nhwc(
m2, m2.add_parameter("x", {migraphx::shape::float_type, {1, 8, 16, 16}}));
auto w =
m2.add_literal(migraphx::generate_literal(
{migraphx::shape::float_type, {8, 8, 1, 1}}));
auto conv = m2.add_instruction(
migraphx::make_op("convolution"),
x,
add_layout_nhwc(m2, w));
auto conv_layout = m2.add_instruction(migraphx::make_op("convolution"), conv, add_layout_nhwc(m2, w));
auto w =
m2.add_literal(migraphx::generate_literal({migraphx::shape::float_type, {8, 8, 1, 1}}));
auto conv = m2.add_instruction(migraphx::make_op("convolution"), x, add_layout_nhwc(m2, w));
auto conv_layout =
m2.add_instruction(migraphx::make_op("convolution"), conv, add_layout_nhwc(m2, w));
m2.add_instruction(layout(), conv_layout);
}
EXPECT(m1.sort() == m2.sort());
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment