Commit b4266bea authored by Jayaram Bobba's avatar Jayaram Bobba

Style fix

parent 2b0cd2fd
......@@ -2372,7 +2372,7 @@ void runtime::cpu::CPU_Emitter::EmitAvgPoolBackprop(
auto delta_shape = args[0].get_shape();
auto delta_rank = delta_shape.size();
auto out_shape = out[0].get_shape();
if (delta_rank == 4 && apb->get_window_shape().size() == 2 &&
args[0].get_element_type() == element::f32)
{
......@@ -2382,13 +2382,14 @@ void runtime::cpu::CPU_Emitter::EmitAvgPoolBackprop(
writer.indent++;
writer << "engine cpu_engine = engine(engine::cpu, 0);\n";
writer << "memory::desc input_data_desc = memory::desc({" << join(delta_shape) << "}, " << et
<< ", memory::format::nchw);\n";
writer << "memory::desc input_data_desc = memory::desc({" << join(delta_shape) << "}, "
<< et << ", memory::format::nchw);\n";
writer << "memory::desc result_desc = memory::desc({" << join(out_shape) << "}, " << et
<< ", memory::format::nchw);\n";
writer << "memory input_data = memory({input_data_desc, cpu_engine}, " << args[0].get_name()
<< ");\n";
writer << "memory result = memory({result_desc, cpu_engine}, " << out[0].get_name() << ");\n";
writer << "memory result = memory({result_desc, cpu_engine}, " << out[0].get_name()
<< ");\n";
// Dummy forward primitive descriptor to keep MKLDNN happy
writer << "pooling_forward::primitive_desc fwd_pd = pooling_forward::primitive_desc("
<< "{prop_kind::forward, algorithm::pooling_avg_exclude_padding, "
......@@ -2409,17 +2410,19 @@ void runtime::cpu::CPU_Emitter::EmitAvgPoolBackprop(
<< "s.submit({avg_pooling}).wait();\n";
writer.indent--;
writer << "}\n";
} else {
}
else
{
writer << "kernel::avg_pool_backprop<" << out[0].get_type() << ">(" << args[0].get_name()
<< ",\n";
writer << " " << out[0].get_name() << ",\n";
writer << " {" << join(delta_shape) << "},\n";
writer << " {" << join(out_shape) << "},\n";
writer << " {" << join(apb->get_window_shape()) << "},\n";
writer << " {" << join(apb->get_window_movement_strides()) << "},\n";
writer << " {" << join(apb->get_padding_below()) << "},\n";
writer << " {" << join(apb->get_padding_above()) << "}\n";
writer << " );\n";
writer << " {" << join(delta_shape) << "},\n";
writer << " {" << join(out_shape) << "},\n";
writer << " {" << join(apb->get_window_shape()) << "},\n";
writer << " {" << join(apb->get_window_movement_strides()) << "},\n";
writer << " {" << join(apb->get_padding_below()) << "},\n";
writer << " {" << join(apb->get_padding_above()) << "}\n";
writer << " );\n";
}
}
......
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