提交 0f94c1ac 编写于 作者: M minqiyang

Polish code

test=develop
上级 c09a3790
...@@ -40,7 +40,7 @@ struct EmbeddingVSumFunctor { ...@@ -40,7 +40,7 @@ struct EmbeddingVSumFunctor {
int64_t row_number = table_t->dims()[0]; int64_t row_number = table_t->dims()[0];
int64_t row_width = table_t->dims()[1]; int64_t row_width = table_t->dims()[1];
int64_t last_dim = output_t->dims()[1]; int64_t last_dim = output_t->dims()[1];
int64_t *ids = ids_t->mutable_data<int64_t>(platform::CPUPlace()); const int64_t *ids = ids_t->data<int64_t>();
auto ids_lod = ids_t->lod()[0]; auto ids_lod = ids_t->lod()[0];
int64_t ids_count = ids_t->numel() / ids_lod.back(); int64_t ids_count = ids_t->numel() / ids_lod.back();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册