Polish code

test=develop
revert-15207-remove_op_handle_lock_and_fix_var
minqiyang 6 years ago
parent c09a379015
commit 0f94c1ac14

@ -40,7 +40,7 @@ struct EmbeddingVSumFunctor {
int64_t row_number = table_t->dims()[0];
int64_t row_width = table_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];
int64_t ids_count = ids_t->numel() / ids_lod.back();

Loading…
Cancel
Save