We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9ff0db3 commit f27ff4dCopy full SHA for f27ff4d
paddle/gserver/tests/test_LayerGrad.cpp
@@ -187,15 +187,13 @@ TEST(Layer, BilinearInterpLayer) {
187
bilinear->set_img_size_y(32);
188
bilinear->set_num_channels(4);
189
190
- bilinear->set_out_size_x(32);
191
- bilinear->set_out_size_y(32);
192
- testLayerGrad(config, "bilinear_interp", 10, false, false);
193
- testLayerGrad(config, "bilinear_interp", 10, false, true);
194
-
195
- bilinear->set_out_size_x(64);
196
- bilinear->set_out_size_y(64);
197
198
+ for (auto useGpu : {false, true}) {
+ for (auto outSize : {32, 64}) {
+ bilinear->set_out_size_x(outSize);
+ bilinear->set_out_size_y(outSize);
+ testLayerGrad(config, "bilinear_interp", 10, false, useGpu);
+ }
199
}
200
201
TEST(Layer, concat) {
0 commit comments