@@ -92,12 +92,12 @@ class GRUUnitKernel : public framework::OpKernel<T> {
92
92
gate_data, frame_size * 3 );
93
93
94
94
// calculate activited gate
95
- Eigen::array<int , 2 > extents = { batch_size, frame_size};
96
- Eigen::array<int , 2 > u_offsets = { 0 , 0 };
95
+ Eigen::array<int , 2 > extents{{ batch_size, frame_size} };
96
+ Eigen::array<int , 2 > u_offsets{{ 0 , 0 } };
97
97
ActCompute (context.Attr <int >(" gate_activation" ), place,
98
98
g.slice (u_offsets, extents), g.slice (u_offsets, extents));
99
99
auto u = g.slice (u_offsets, extents); // update gate
100
- Eigen::array<int , 2 > r_offsets = { 0 , frame_size};
100
+ Eigen::array<int , 2 > r_offsets{{ 0 , frame_size} };
101
101
ActCompute (context.Attr <int >(" gate_activation" ), place,
102
102
g.slice (r_offsets, extents), g.slice (r_offsets, extents));
103
103
auto r = g.slice (r_offsets, extents); // reset gate
@@ -107,7 +107,7 @@ class GRUUnitKernel : public framework::OpKernel<T> {
107
107
weight_data + frame_size * frame_size * 2 , frame_size, 1 ,
108
108
gate_data + frame_size * 2 , frame_size * 3 );
109
109
110
- Eigen::array<int , 2 > c_offsets = { 0 , frame_size * 2 };
110
+ Eigen::array<int , 2 > c_offsets{{ 0 , frame_size * 2 } };
111
111
ActCompute (context.Attr <int >(" activation" ), place,
112
112
g.slice (c_offsets, extents), g.slice (c_offsets, extents));
113
113
auto c = g.slice (c_offsets, extents); // output candidate
@@ -171,12 +171,12 @@ class GRUUnitGradKernel : public framework::OpKernel<T> {
171
171
int batch_size = input->dims ()[0 ];
172
172
int frame_size = hidden_prev->dims ()[1 ];
173
173
174
- Eigen::array<int , 2 > extents = { batch_size, frame_size};
175
- Eigen::array<int , 2 > u_offsets = { 0 , 0 };
174
+ Eigen::array<int , 2 > extents{{ batch_size, frame_size} };
175
+ Eigen::array<int , 2 > u_offsets{{ 0 , 0 } };
176
176
auto u = g.slice (u_offsets, extents); // update gate
177
- Eigen::array<int , 2 > r_offsets = { 0 , frame_size};
177
+ Eigen::array<int , 2 > r_offsets{{ 0 , frame_size} };
178
178
auto r = g.slice (r_offsets, extents); // reset gate
179
- Eigen::array<int , 2 > c_offsets = { 0 , frame_size * 2 };
179
+ Eigen::array<int , 2 > c_offsets{{ 0 , frame_size * 2 } };
180
180
auto c = g.slice (c_offsets, extents); // output candidate
181
181
182
182
// backward for unactivated update gate
0 commit comments