Commit aa95521b authored by Paul's avatar Paul
Browse files

Formatting

parent 5f37fd2d
...@@ -60,8 +60,10 @@ inline convolution_descriptor make_conv(const rtg::convolution& op) ...@@ -60,8 +60,10 @@ inline convolution_descriptor make_conv(const rtg::convolution& op)
inline pooling_descriptor make_pooling(const rtg::pooling& op) inline pooling_descriptor make_pooling(const rtg::pooling& op)
{ {
miopenPoolingMode_t mode; miopenPoolingMode_t mode;
if(op.mode == "max") mode = miopenPoolingMax; if(op.mode == "max")
else mode = miopenPoolingAverage; mode = miopenPoolingMax;
else
mode = miopenPoolingAverage;
auto p = make_obj<pooling_descriptor>(&miopenCreatePoolingDescriptor); auto p = make_obj<pooling_descriptor>(&miopenCreatePoolingDescriptor);
miopenSet2dPoolingDescriptor(p.get(), miopenSet2dPoolingDescriptor(p.get(),
mode, mode,
......
...@@ -180,11 +180,8 @@ struct miopen_apply ...@@ -180,11 +180,8 @@ struct miopen_apply
auto pd = make_pooling(op); auto pd = make_pooling(op);
auto output = insert_allocation(ins, ins->result); auto output = insert_allocation(ins, ins->result);
prog->replace_instruction(ins, prog->replace_instruction(
miopen_pooling{op, std::move(pd)}, ins, miopen_pooling{op, std::move(pd)}, handle, ins->arguments.at(0), output);
handle,
ins->arguments.at(0),
output);
} }
void apply_activation(instruction_ref ins) void apply_activation(instruction_ref ins)
......
...@@ -93,7 +93,8 @@ struct test_conv_pooling ...@@ -93,7 +93,8 @@ struct test_conv_pooling
} }
}; };
int main() { int main()
{
verify_program<test_conv_relu>(); verify_program<test_conv_relu>();
verify_program<test_conv_pooling>(); verify_program<test_conv_pooling>();
} }
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