| 25 | using singa::Flatten; |
| 26 | using singa::Shape; |
| 27 | TEST(Flatten, Setup) { |
| 28 | Flatten flt; |
| 29 | // EXPECT_EQ("Flatten", flt.layer_type()); |
| 30 | |
| 31 | singa::LayerConf conf; |
| 32 | singa::FlattenConf *flattenconf = conf.mutable_flatten_conf(); |
| 33 | flattenconf->set_axis(1); |
| 34 | |
| 35 | flt.Setup(Shape{2}, conf); |
| 36 | EXPECT_EQ(1, flt.Axis()); |
| 37 | } |
| 38 | |
| 39 | TEST(Flatten, ForwardCPU) { |
| 40 | const float x[] = {1.f, 2.f, 3.f, -2.f, -3.f, -4.f, |