@@ -179,25 +179,23 @@ TEST(Layer, BilinearInterpLayer) {
179
179
TestConfig config;
180
180
config.layerConfig .set_type (" bilinear_interp" );
181
181
config.biasSize = 0 ;
182
-
183
182
config.inputDefs .push_back ({INPUT_DATA, " layer_0" , 4096 , 0 });
184
183
185
- for (auto useGpu : {false , true }) {
186
- for (auto out_size : {32 , 64 }) {
187
- LOG (INFO) << " out_size_x=" << out_size
188
- << " out_size_y=" << out_size;
189
- LayerInputConfig* input
190
- = config.layerConfig .add_inputs ();
191
- BilinearInterpConfig* bilinear
192
- = input->mutable_bilinear_interp_conf ();
193
- bilinear->set_img_size_x (32 );
194
- bilinear->set_img_size_y (32 );
195
- bilinear->set_num_channels (4 );
196
- bilinear->set_out_size_x (out_size);
197
- bilinear->set_out_size_y (out_size);
198
- testLayerGrad (config, " bilinear_interp" , 10 , false , useGpu);
199
- }
200
- }
184
+ LayerInputConfig* input = config.layerConfig .add_inputs ();
185
+ BilinearInterpConfig* bilinear = input->mutable_bilinear_interp_conf ();
186
+ bilinear->set_img_size_x (32 );
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
+ testLayerGrad (config, " bilinear_interp" , 10 , false , false );
198
+ testLayerGrad (config, " bilinear_interp" , 10 , false , true );
201
199
}
202
200
203
201
TEST (Layer, concat) {
0 commit comments