| 125 | } // namespace |
| 126 | |
| 127 | void TensorSliceWriteTestHelper::GetData(TensorSliceReader::Table* table, |
| 128 | const string& name, |
| 129 | const TensorSlice& slice, |
| 130 | SavedSlice* ss) { |
| 131 | string key = EncodeTensorNameSlice(name, slice); |
| 132 | string value; |
| 133 | EXPECT_TRUE(table->Get(key, &value)); |
| 134 | SavedTensorSlices sts; |
| 135 | EXPECT_TRUE(ParseProtoUnlimited(&sts, value)); |
| 136 | EXPECT_FALSE(sts.has_meta()); |
| 137 | *ss = sts.data(); |
| 138 | EXPECT_EQ(name, ss->name()); |
| 139 | TensorSlice slice2(ss->slice()); |
| 140 | EXPECT_EQ(slice.DebugString(), slice2.DebugString()); |
| 141 | } |
| 142 | |
| 143 | void TensorSliceWriteTestHelper::CheckEntries(const string& fname) { |
| 144 | TensorSliceReader::Table* tptr; |
no test coverage detected