@@ -30,23 +30,23 @@ using std::endl;
30
30
31
31
void CreateInput (LoDTensor* ids, LoDTensor* scores) {
32
32
LoD lod;
33
- vector<size_t > level0{0 , 2 , 4 };
34
- vector<size_t > level1{0 , 1 , 2 , 3 , 4 };
33
+ vector<size_t > level0 ( {0 , 2 , 4 }) ;
34
+ vector<size_t > level1 ( {0 , 1 , 2 , 3 , 4 }) ;
35
35
lod.push_back (level0);
36
36
lod.push_back (level1);
37
37
ids->set_lod (lod);
38
38
scores->set_lod (lod);
39
39
40
- auto dims = framework::make_ddim (vector<int64_t >{4 , 3 });
40
+ auto dims = framework::make_ddim (vector<int64_t >( {4 , 3 }) );
41
41
ids->Resize (dims);
42
42
scores->Resize (dims);
43
43
CPUPlace place;
44
44
45
45
auto * ids_data = ids->mutable_data <int64_t >(place);
46
46
auto * scores_data = scores->mutable_data <float >(place);
47
- vector<int64_t > _ids{4 , 2 , 5 , 2 , 1 , 3 , 3 , 5 , 2 , 8 , 2 , 1 };
48
- vector<float > _scores{0 .5f , 0 .3f , 0 .2f , 0 .6f , 0 .3f , 0 .1f ,
49
- 0 .9f , 0 .5f , 0 .1f , 0 .7f , 0 .5f , 0 .1f };
47
+ vector<int64_t > _ids ( {4 , 2 , 5 , 2 , 1 , 3 , 3 , 5 , 2 , 8 , 2 , 1 }) ;
48
+ vector<float > _scores ( {0 .5f , 0 .3f , 0 .2f , 0 .6f , 0 .3f , 0 .1f ,
49
+ 0 .9f , 0 .5f , 0 .1f , 0 .7f , 0 .5f , 0 .1f }) ;
50
50
51
51
for (int i = 0 ; i < 12 ; i++) {
52
52
ids_data[i] = _ids[i];
@@ -79,8 +79,8 @@ TEST(DISABLED_beam_search_op, run) {
79
79
80
80
ASSERT_EQ (sids.lod (), sscores.lod ());
81
81
82
- vector<int > tids{4 , 2 , 3 , 8 };
83
- vector<float > tscores{0 .5f , 0 .6f , 0 .9f , 0 .7f };
82
+ vector<int > tids ( {4 , 2 , 3 , 8 }) ;
83
+ vector<float > tscores ( {0 .5f , 0 .6f , 0 .9f , 0 .7f }) ;
84
84
85
85
for (int i = 0 ; i < 4 ; i++) {
86
86
ASSERT_EQ (tids[i], sids.data <int64_t >()[i]);
0 commit comments