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

formatting

parent d0afbcaf
......@@ -177,7 +177,7 @@ struct tf_parser
parse_biasadd(const std::string&, const attribute_map&, std::vector<instruction_ref> args)
{
uint64_t axis = 1; // assume output of previous layer is in NCHW (broadcast on channel)
auto l0 = args[0];
auto l0 = args[0];
// otherwise, if the input is a parameter to the graph, then first insert transpose
if(l0->name() == "@param")
{
......
......@@ -20,7 +20,7 @@ TEST_CASE(add_test)
TEST_CASE(add_bcast_test)
{
migraphx::program p;
migraphx::shape s0{migraphx::shape::float_type, {2, 3}};
auto l0 = p.add_parameter("0", s0);
......@@ -38,10 +38,10 @@ TEST_CASE(biasadd_test)
migraphx::program p;
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 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);
auto l0 = p.add_parameter("0", s0);
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);
auto prog = migraphx::parse_tf("biasadd_test.pb", true);
......@@ -66,13 +66,13 @@ TEST_CASE(pooling_test)
migraphx::op::pooling max_pool_op{"max"};
avg_pool_op.padding_mode = migraphx::op::padding_mode_t::valid;
max_pool_op.padding_mode = migraphx::op::padding_mode_t::valid;
avg_pool_op.stride = {2, 2};
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);
avg_pool_op.stride = {2, 2};
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);
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