提交 783cc27a 编写于 作者: L liuqi

Add all kernel size to conv2d test.

上级 4727af0d
......@@ -240,8 +240,8 @@ TEST_F(Conv2dOpTest, ConvNxNS12) {
};
for (int kernel_size : {3}) { // TODO(liu1i10) 3x3
for (int stride : {1}) {
for (int kernel_size : {1, 3}) {
for (int stride : {1, 2}) {
func(kernel_size, kernel_size, stride, stride, VALID);
func(kernel_size, kernel_size, stride, stride, SAME);
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册