rename parameter

wangkuiyi-patch-2
Yancey1989 7 years ago
parent 2917a75b7a
commit b920b51686

@ -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;

@ -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.

Loading…
Cancel
Save