Commit da0ddd48 authored by rocking's avatar rocking
Browse files

Fix order of parameter in example

parent 58d84615
......@@ -84,9 +84,9 @@ bool pool3d_bwd_test(bool do_verification,
static_cast<DInDataType*>(din_device_buf.GetDeviceBuffer()),
{N, C, Do, Ho, Wo},
{N, C, Di, Hi, Wi},
window_lengths,
{Do * C * Ho * Wo, 1, C * Ho * Wo, Wo * C, C},
{Di * C * Hi * Wi, 1, C * Hi * Wi, Wi * C, C},
window_lengths,
window_strides,
window_dilations,
dinput_left_pads,
......
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