@@ -107,7 +107,8 @@ template <typename T>
107
107
class LastSeqPoolFunctor {
108
108
public:
109
109
void operator ()(const platform::CPUDeviceContext& context,
110
- const framework::LoDTensor& input, framework::Tensor* output) {
110
+ const framework::LoDTensor& input,
111
+ framework::Tensor* output) {
111
112
// Create pointers to input and output data
112
113
auto * in_data = input.data <T>();
113
114
auto * out_data = output->data <T>();
@@ -124,15 +125,16 @@ class LastSeqPoolFunctor {
124
125
// Copy the last item of sequence to output
125
126
std::memcpy (out_data, (in_data - item_size), item_size * sizeof (T));
126
127
out_data += item_size;
127
- }
128
+ }
128
129
}
129
130
};
130
131
131
132
template <typename T>
132
133
class FirstSeqPoolFunctor {
133
134
public:
134
135
void operator ()(const platform::CPUDeviceContext& context,
135
- const framework::LoDTensor& input, framework::Tensor* output) {
136
+ const framework::LoDTensor& input,
137
+ framework::Tensor* output) {
136
138
// Create pointers to input and output data
137
139
auto * in_data = input.data <T>();
138
140
auto * out_data = output->data <T>();
@@ -149,7 +151,7 @@ class FirstSeqPoolFunctor {
149
151
// Point to the next sequence
150
152
in_data += seq_len * item_size;
151
153
out_data += item_size;
152
- }
154
+ }
153
155
}
154
156
};
155
157
@@ -176,8 +178,6 @@ class SequencePoolFunctor<platform::CPUDeviceContext, T> {
176
178
first_pool (context, input, output);
177
179
return ;
178
180
}
179
-
180
-
181
181
auto lod = input.lod ()[0 ];
182
182
auto & place = *context.eigen_device ();
183
183
for (int i = 0 ; i < static_cast <int >(lod.size ()) - 1 ; ++i) {
0 commit comments