Commit bad02557 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

refine the unit test

parent 7ac9c87b
......@@ -10,7 +10,7 @@ struct test_mul_lrn : verify_program<test_mul_lrn>
{
migraphx::program p;
auto* mm = p.get_main_module();
migraphx::shape s{migraphx::shape::half_type, {1, 5, 2, 2}};
migraphx::shape s{migraphx::shape::half_type, {1, 3, 224, 224}};
auto x = mm->add_parameter("x", s);
std::vector<float> vec(s.elements(), 500.0f);
auto l = mm->add_literal(migraphx::literal(s, vec));
......@@ -18,7 +18,7 @@ struct test_mul_lrn : verify_program<test_mul_lrn>
auto y = mm->add_instruction(migraphx::make_op("relu"), xl);
mm->add_instruction(
migraphx::make_op("lrn",
{{"alpha", 0.0001}, {"beta", 0.75}, {"bias", 1.0}, {"size", 5}}),
{{"alpha", 0.0001}, {"beta", 0.75}, {"bias", 1.0}, {"size", 50}}),
y);
return p;
}
......
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