提交 2b1f21a5 编写于 作者: Y Yu Yang 提交者: QI JUN

Fix MacOS Compile (#5217)

上级 cdc700bb
...@@ -48,7 +48,7 @@ class SeqExpandKernel : public framework::OpKernel<T> { ...@@ -48,7 +48,7 @@ class SeqExpandKernel : public framework::OpKernel<T> {
x_t(x_data, 1, element_len); x_t(x_data, 1, element_len);
Eigen::TensorMap<Eigen::Tensor<T, 2, Eigen::RowMajor, Eigen::DenseIndex>> Eigen::TensorMap<Eigen::Tensor<T, 2, Eigen::RowMajor, Eigen::DenseIndex>>
out_t(out_data, scale, element_len); out_t(out_data, scale, element_len);
Eigen::array<int, 2> cast({scale, 1}); Eigen::array<int, 2> cast({{scale, 1}});
out_t.device(place) = x_t.broadcast(cast); out_t.device(place) = x_t.broadcast(cast);
x_data += element_len; x_data += element_len;
out_data += element_len * scale; out_data += element_len * scale;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册