From b920b51686d77db07fde4449e2225ae3e363b3f7 Mon Sep 17 00:00:00 2001 From: Yancey1989 Date: Wed, 18 Apr 2018 13:46:52 +0800 Subject: [PATCH] rename parameter --- paddle/fluid/framework/selected_rows.cc | 4 ++-- paddle/fluid/framework/selected_rows.h | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/framework/selected_rows.cc b/paddle/fluid/framework/selected_rows.cc index ec82611d2e0..b1837ca3c42 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 a6e96872515..f329ae8939d 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. -- GitLab