| 112 | |
| 113 | template <typename T> |
| 114 | void ExpectNext(BundleReader* reader, const Tensor& expected_val) { |
| 115 | EXPECT_TRUE(reader->Valid()); |
| 116 | reader->Next(); |
| 117 | TF_ASSERT_OK(reader->status()); |
| 118 | Tensor val; |
| 119 | TF_ASSERT_OK(reader->ReadCurrent(&val)); |
| 120 | test::ExpectTensorEqual<T>(val, expected_val); |
| 121 | } |
| 122 | |
| 123 | std::vector<string> AllTensorKeys(BundleReader* reader) { |
| 124 | std::vector<string> ret; |
nothing calls this directly
no test coverage detected