| 183 | } |
| 184 | |
| 185 | void TensorSliceReader::LoadAllShards() const { |
| 186 | VLOG(1) << "Loading all shards for " << filepattern_; |
| 187 | for (size_t i = 0; i < fnames_.size() && status_.ok(); ++i) { |
| 188 | LoadShard(i); |
| 189 | } |
| 190 | all_shards_loaded_ = true; |
| 191 | } |
| 192 | |
| 193 | const TensorSliceSet* TensorSliceReader::FindTensorSlice( |
| 194 | const string& name, const TensorSlice& slice, |