提交 7aa8b2cc 编写于 作者: Q Qiao Longfei

optimize code

上级 bad0c27e
...@@ -70,7 +70,7 @@ struct TensorFillVisitor { ...@@ -70,7 +70,7 @@ struct TensorFillVisitor {
template <typename T> template <typename T>
void apply() const { void apply() const {
// TODO(Yancey1989): support other place // TODO(qiao): support other place
platform::CPUPlace cpu; platform::CPUPlace cpu;
auto* tensor_data = dst_->mutable_data<T>(cpu); auto* tensor_data = dst_->mutable_data<T>(cpu);
auto* start = tensor_data + dst_offset_; auto* start = tensor_data + dst_offset_;
......
...@@ -94,7 +94,7 @@ class LookupSparseTableOpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -94,7 +94,7 @@ class LookupSparseTableOpMaker : public framework::OpProtoAndCheckerMaker {
.SetDefault(true); .SetDefault(true);
AddAttr<bool>("is_test", AddAttr<bool>("is_test",
"In test mode, lookup_sparse_table will " "In test mode, lookup_sparse_table will "
"return a default value for unknown id") "return a 0 for unknown id")
.SetDefault(false); .SetDefault(false);
AddComment(R"DOC( AddComment(R"DOC(
Lookup Sprase Tablel Operator. Lookup Sprase Tablel Operator.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册