diff --git a/paddle/fluid/framework/selected_rows.cc b/paddle/fluid/framework/selected_rows.cc index ec82611d2e0b221d078e73bf7a16ae9f960ee8f4..b1837ca3c425a5bd580f4403c24fc3f2b0b91972 100644 --- a/paddle/fluid/framework/selected_rows.cc +++ b/paddle/fluid/framework/selected_rows.cc @@ -122,7 +122,7 @@ bool SelectedRows::HasKey(int64_t key) const { } bool SelectedRows::Get(int64_t key, framework::Tensor* value, - int64_t row) const { + int64_t offset) const { int64_t index = Index(key); PADDLE_ENFORCE_GE(index, 0, "The key should be exists in the Table."); PADDLE_ENFORCE(value->IsInitialized(), @@ -138,7 +138,7 @@ bool SelectedRows::Get(int64_t key, framework::Tensor* value, framework::VisitDataType( framework::ToDataType(value_->type()), - TensorCopyVisitor(cpu, value, row * value_width, *value_.get(), + TensorCopyVisitor(cpu, value, offset * value_width, *value_.get(), index * value_width, value_width)); return true; diff --git a/paddle/fluid/framework/selected_rows.h b/paddle/fluid/framework/selected_rows.h index a6e968725150ad96067fbfccacb546e9fdf69c77..f329ae8939d996363d469c593903f8bfb0b2abe3 100644 --- a/paddle/fluid/framework/selected_rows.h +++ b/paddle/fluid/framework/selected_rows.h @@ -81,7 +81,7 @@ class SelectedRows { * @return true if the Get operation successed. */ - bool Get(int64_t key, framework::Tensor* tensor, int64_t row = 0) const; + bool Get(int64_t key, framework::Tensor* tensor, int64_t offset = 0) const; /* * @brief Set a key-value pair into the table.