提交 ee414010 编写于 作者: X xzl

add more dialtion test

上级 d746e49f
......@@ -79,9 +79,13 @@ void Convolution(const std::string& conv1,
if (outputChannels < inputChannels) continue;
for (size_t stride : {1, 2}) {
for (size_t padding : {0, 1}) {
for (size_t dilation : {1}) {
for (size_t dilation : {1, 3}) {
if (padding >= filterSize) break;
if ((conv1 == "NaiveConv-CPU" || conv2 == "NaiveConv-CPU") &&
dilation > 1)
break;
// NNPACK only supports stride = 1 if batchSize > 1
if ((conv1 == "NNPACKConv-CPU" ||
conv2 == "NNPACKConv-CPU") &&
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册