| 97 | } |
| 98 | |
| 99 | std::vector<string> AllTensorKeys(BundleReader* reader) { |
| 100 | std::vector<string> ret; |
| 101 | reader->Seek(kHeaderEntryKey); |
| 102 | reader->Next(); |
| 103 | for (; reader->Valid(); reader->Next()) { |
| 104 | //ret.push_back(reader->key().ToString()); |
| 105 | ret.push_back(std::string(reader->key())); |
| 106 | } |
| 107 | return ret; |
| 108 | } |
| 109 | |
| 110 | TEST(EmbeddingVariableTest, TestEmptyEV) { |
| 111 | int64 value_size = 8; |