new TensorSliceReader::VarToDataTypeMap);
  v2_reader_->Seek(kHeaderEntryKey);
  for (v2_reader_->Next(); v2_reader_->Valid(); v2_reader_->Next()) {
    if (filtered_keys.count(string(v2_reader_->key())) > 0) continue;
    CHECK(entry.ParseFromArray(v2_reader_->value().data(),
                               v2_reader_->value().size()))
        << entry.InitializationErrorString();
    string key(v2_reader_->key());

  ASSERT_NE(func_, nullptr);

  // Verify that FunctionDef has 2 attributes, "v1" and "v2".
  ASSERT_EQ(func_->record->fdef().signature().attr().size(), 2);
  EXPECT_EQ(func_->record->fdef().signature().attr(0).name(), "v1");
  EXPECT_EQ(func_->record->fdef().signature().attr(0).type(), "int");
  EXPECT_EQ(func_->record->fdef().signature().attr(1).name(), "v2");
  EXPECT_EQ(func_->record->fdef().signature().attr(1).type(), "int");
}