Skip to content

Commit 642593a

Browse files
authored
[Embedding] Add empty check when using SSDIterator to traverse SSDHashKV. (#292)
1 parent a7c9ddd commit 642593a

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tensorflow/core/framework/embedding/ssd_hashkv.h

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -147,8 +147,10 @@ class SSDIterator : public Iterator {
147147
virtual void SeekToFirst() {
148148
curr_file_ = 0;
149149
curr_vec_ = 0;
150-
int64 f_id = file_id_vec_[curr_file_];
151-
emb_files_[f_id]->Map();
150+
if (file_id_vec_.size() > 0) {
151+
int64 f_id = file_id_vec_[curr_file_];
152+
emb_files_[f_id]->Map();
153+
}
152154
}
153155
virtual void Next() {
154156
curr_vec_++;

0 commit comments

Comments
 (0)