Function TEST

test/singa/test_snapshot.cc:40–47

const double double_data[] = {0.2, 0.4, 0.6, 0.8};

TEST(Snapshot, WriteTest) {
  singa::Snapshot snapshot(prefix, singa::Snapshot::kWrite);
  singa::Tensor param_1(singa::Shape{4}), param_2(singa::Shape{2, 2});
  param_1.CopyDataFromHostPtr(param_1_data, 4);
  param_2.CopyDataFromHostPtr(param_2_data, 4);
  snapshot.Write("Param_1", param_1);
  snapshot.Write("Param_2", param_2);
}

TEST(Snapshot, ReadTest) {
  singa::Snapshot snapshot(prefix, singa::Snapshot::kRead);