Function TEST

test/singa/test_slice.cc:26–38

using singa::Shape;
TEST(Slice, Setup) {
  singa::LayerConf conf;
  conf.set_type("singa_slice");
  auto slice_conf = conf.mutable_slice_conf();
  slice_conf->set_axis(1);
  slice_conf->add_slice_point(2);
  singa::Slice layer;
  layer.Setup({3u}, conf);
  auto s1 = layer.GetOutputSampleShape(0);
  EXPECT_EQ(s1[0], 2u);
  auto s2 = layer.GetOutputSampleShape(1);
  EXPECT_EQ(s2[0], 1u);
}

void ForwardSliceRowTest(std::shared_ptr<singa::Device> dev) {
  size_t a = 2u, b = 1u, c = 3u;