@@ -32,28 +32,6 @@ P_DECLARE_double(checkgrad_eps);
32
32
P_DECLARE_bool (thread_local_rand_use_global_seed);
33
33
P_DECLARE_bool (prev_batch_state);
34
34
35
- TEST (Layer, BilinearInterpLayer) {
36
- TestConfig config;
37
- config.layerConfig .set_type (" bilinear_interp" );
38
- config.biasSize = 0 ;
39
-
40
- config.inputDefs .push_back ({INPUT_DATA, " layer_0" , 4096 , 0 });
41
- LayerInputConfig* input = config.layerConfig .add_inputs ();
42
- BilinearInterpConfig* bilinear = input->mutable_bilinear_interp_conf ();
43
-
44
- bilinear->set_img_size_x (32 );
45
- bilinear->set_img_size_y (32 );
46
- bilinear->set_num_channels (4 );
47
-
48
- for (auto useGpu : {false , true }) {
49
- for (auto out_size : {32 , 64 , 128 }) {
50
- bilinear->set_out_size_x (out_size);
51
- bilinear->set_out_size_y (out_size);
52
- testLayerGrad (config, " bilinear_interp" , 10 , false , useGpu);
53
- }
54
- }
55
- }
56
-
57
35
TEST (Operator, dot_mul) {
58
36
TestConfig config;
59
37
config.layerConfig .set_size (10 );
@@ -197,6 +175,31 @@ TEST(Projection, conv) {
197
175
}
198
176
#endif
199
177
178
+ TEST (Layer, BilinearInterpLayer) {
179
+ TestConfig config;
180
+ config.layerConfig .set_type (" bilinear_interp" );
181
+ config.biasSize = 0 ;
182
+
183
+ config.inputDefs .push_back ({INPUT_DATA, " layer_0" , 4096 , 0 });
184
+
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
+ }
201
+ }
202
+
200
203
TEST (Layer, concat) {
201
204
TestConfig config;
202
205
config.biasSize = 0 ;
0 commit comments