Commit f4d3089c authored by Nick Korovaiko's avatar Nick Korovaiko Committed by Matthew Brookhart

bprop fixes for avg/max pool in cpu_emitter (#455)

parent 721fdcbd
......@@ -2368,8 +2368,8 @@ void runtime::cpu::CPU_Emitter::EmitAvgPoolBackprop(
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 << " true);\n";
writer << " {" << join(apb->get_padding_above()) << "}\n";
writer << " );\n";
}
void runtime::cpu::CPU_Emitter::EmitMaxPoolBackprop(
......@@ -2392,8 +2392,8 @@ void runtime::cpu::CPU_Emitter::EmitMaxPoolBackprop(
writer << " {" << join(mpb->get_window_shape()) << "},\n";
writer << " {" << join(mpb->get_window_movement_strides()) << "},\n";
writer << " {" << join(mpb->get_padding_below()) << "},\n";
writer << " {" << join(mpb->get_padding_above()) << "},\n";
writer << " true);\n";
writer << " {" << join(mpb->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