提交 bfd663c2 编写于 作者: D Dmitry Kurtaev

Add a test for grouped deconvolution from ONNX

上级 5c7fe0fe
......@@ -1217,12 +1217,8 @@ public:
int dims[] = {inputs[0][0], outCn, outH, outW};
outputs.resize(inputs.size(), shape(dims, 4));
internals.push_back(MatShape());
if (!is1x1())
internals[0] = computeColRowShape(inputs[0], outputs[0]);
if (hasBias())
internals.push_back(shape(1, outH*outW));
internals.push_back(computeColRowShape(inputs[0], outputs[0]));
return false;
}
......
......@@ -72,6 +72,7 @@ TEST_P(Test_ONNX_layers, Deconvolution)
{
testONNXModels("deconvolution");
testONNXModels("two_deconvolution");
testONNXModels("deconvolution_group");
}
TEST_P(Test_ONNX_layers, Dropout)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册