提交 76553c5a 编写于 作者: B bingyanghuang

fix travis-ci

上级 83394bab
...@@ -107,7 +107,8 @@ template <typename T> ...@@ -107,7 +107,8 @@ template <typename T>
class LastSeqPoolFunctor { class LastSeqPoolFunctor {
public: public:
void operator()(const platform::CPUDeviceContext& context, void operator()(const platform::CPUDeviceContext& context,
const framework::LoDTensor& input, framework::Tensor* output) { const framework::LoDTensor& input,
framework::Tensor* output) {
// Create pointers to input and output data // Create pointers to input and output data
auto* in_data = input.data<T>(); auto* in_data = input.data<T>();
auto* out_data = output->data<T>(); auto* out_data = output->data<T>();
...@@ -124,7 +125,7 @@ class LastSeqPoolFunctor { ...@@ -124,7 +125,7 @@ class LastSeqPoolFunctor {
// Copy the last item of sequence to output // Copy the last item of sequence to output
std::memcpy(out_data, (in_data - item_size), item_size * sizeof(T)); std::memcpy(out_data, (in_data - item_size), item_size * sizeof(T));
out_data += item_size; out_data += item_size;
} }
} }
}; };
...@@ -132,7 +133,8 @@ template <typename T> ...@@ -132,7 +133,8 @@ template <typename T>
class FirstSeqPoolFunctor { class FirstSeqPoolFunctor {
public: public:
void operator()(const platform::CPUDeviceContext& context, void operator()(const platform::CPUDeviceContext& context,
const framework::LoDTensor& input, framework::Tensor* output) { const framework::LoDTensor& input,
framework::Tensor* output) {
// Create pointers to input and output data // Create pointers to input and output data
auto* in_data = input.data<T>(); auto* in_data = input.data<T>();
auto* out_data = output->data<T>(); auto* out_data = output->data<T>();
...@@ -149,7 +151,7 @@ class FirstSeqPoolFunctor { ...@@ -149,7 +151,7 @@ class FirstSeqPoolFunctor {
// Point to the next sequence // Point to the next sequence
in_data += seq_len * item_size; in_data += seq_len * item_size;
out_data += item_size; out_data += item_size;
} }
} }
}; };
...@@ -176,8 +178,6 @@ class SequencePoolFunctor<platform::CPUDeviceContext, T> { ...@@ -176,8 +178,6 @@ class SequencePoolFunctor<platform::CPUDeviceContext, T> {
first_pool(context, input, output); first_pool(context, input, output);
return; return;
} }
auto lod = input.lod()[0]; auto lod = input.lod()[0];
auto& place = *context.eigen_device(); auto& place = *context.eigen_device();
for (int i = 0; i < static_cast<int>(lod.size()) - 1; ++i) { for (int i = 0; i < static_cast<int>(lod.size()) - 1; ++i) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册