Commit 1bd9d6a0 authored by Khalique's avatar Khalique
Browse files

formatting

parent d0afbcaf
......@@ -39,7 +39,7 @@ TEST_CASE(biasadd_test)
migraphx::shape s0{migraphx::shape::float_type, {1, 1, 1, 500}};
uint64_t axis = 1;
auto l0 = p.add_parameter("0", s0);
auto l1 = p.add_instruction(migraphx::op::transpose{{0,3,1,2}}, l0);
auto l1 = p.add_instruction(migraphx::op::transpose{{0, 3, 1, 2}}, l0);
auto l2 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {500}});
auto l3 = p.add_instruction(migraphx::op::broadcast{axis, l1->get_shape()}, l2);
p.add_instruction(migraphx::op::add{}, l1, l3);
......@@ -70,9 +70,9 @@ TEST_CASE(pooling_test)
max_pool_op.stride = {2, 2};
avg_pool_op.lengths = {2, 2};
max_pool_op.lengths = {2, 2};
auto l1 = p.add_instruction(migraphx::op::transpose{{0,3,1,2}}, l0);
auto l1 = p.add_instruction(migraphx::op::transpose{{0, 3, 1, 2}}, l0);
p.add_instruction(max_pool_op, l1);
auto l2 = p.add_instruction(migraphx::op::transpose{{0,3,1,2}}, l0);
auto l2 = p.add_instruction(migraphx::op::transpose{{0, 3, 1, 2}}, l0);
p.add_instruction(avg_pool_op, l2);
auto prog = migraphx::parse_tf("pooling_test.pb", true);
......
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