提交 40d65a13 编写于 作者: Q Qiao Longfei

optimize code

上级 d37b9797
...@@ -218,7 +218,7 @@ void ReadThread(const std::vector<std::string>& file_list, ...@@ -218,7 +218,7 @@ void ReadThread(const std::vector<std::string>& file_list,
// insert label tensor // insert label tensor
framework::LoDTensor label_tensor; framework::LoDTensor label_tensor;
int64_t* label_tensor_data = label_tensor.mutable_data<int64_t>( auto* label_tensor_data = label_tensor.mutable_data<int64_t>(
framework::make_ddim({1, static_cast<int64_t>(batch_label.size())}), framework::make_ddim({1, static_cast<int64_t>(batch_label.size())}),
platform::CPUPlace()); platform::CPUPlace());
memcpy(label_tensor_data, batch_label.data(), memcpy(label_tensor_data, batch_label.data(),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册