Commit a79ab4d7 authored by Khalique's avatar Khalique
Browse files

formatting

parent 5168b178
......@@ -88,12 +88,7 @@ inline pooling_descriptor make_pooling(const migraph::op::pooling& op)
inline LRN_descriptor make_LRN(const migraph::op::LRN& op)
{
auto ldesc = make_obj<LRN_descriptor>(&miopenCreateLRNDescriptor);
miopenSetLRNDescriptor(ldesc.get(),
miopenLRNCrossChannel,
op.size,
op.alpha,
op.beta,
op.bias);
miopenSetLRNDescriptor(ldesc.get(), miopenLRNCrossChannel, op.size, op.alpha, op.beta, op.bias);
return ldesc;
}
......
......@@ -142,7 +142,8 @@ struct miopen_apply
auto&& op = any_cast<op::LRN>(ins->get_operator());
auto ldesc = make_LRN(op);
auto output = insert_allocation(ins, ins->get_shape());
return prog->replace_instruction(ins, miopen_LRN{std::move(ldesc)}, ins->inputs().at(0), output);
return prog->replace_instruction(
ins, miopen_LRN{std::move(ldesc)}, ins->inputs().at(0), output);
}
instruction_ref apply_relu(instruction_ref ins)
......
......@@ -589,11 +589,10 @@ TEST_CASE(LRN_test)
auto result = p.eval({});
std::vector<float> results_vector(5);
result.visit([&](auto output) { results_vector.assign(output.begin(), output.end()); });
std::vector<float> gold = {-2/1.000075, 1/1.00009, 0/1.000145, 1/1.00009, 2/1.000075};
std::vector<float> gold = {-2 / 1.000075, 1 / 1.00009, 0 / 1.000145, 1 / 1.00009, 2 / 1.000075};
EXPECT(migraph::verify_range(results_vector, gold));
}
TEST_CASE(imagescaler_test)
{
migraph::program 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