| /external/tensorflow/tensorflow/core/util/ |
| tensor_slice_reader_cache.cc | 34 const string& filepattern,
41 return cache_->GetReader(filepattern, std::move(open_function),
54 const string& filepattern,
77 while (still_opening_.find(filepattern) != still_opening_.end()) {
82 if (readers_.find(filepattern) == readers_.end()) {
83 VLOG(1) << "Creating new TensorSliceReader for " << filepattern;
84 still_opening_.insert(filepattern);
89 new TensorSliceReader(filepattern, open_function, preferred_shard));
94 readers_[filepattern] = std::make_pair(*func_ptr, reader);
98 CHECK_EQ(size_t{1}, still_opening_.erase(filepattern));
[all...] |
| tensor_slice_reader_cache.h | 43 const string& filepattern,
58 // Returns the TensorSliceReader corresponding to 'filepattern' and the
60 // TensorSliceReader for the filepattern/open_function combination.
62 const string& filepattern,
|
| tensor_slice_reader.cc | 99 TensorSliceReader::TensorSliceReader(const string& filepattern)
100 : TensorSliceReader(filepattern, OpenTableTensorSliceReader,
103 TensorSliceReader::TensorSliceReader(const string& filepattern,
105 : TensorSliceReader(filepattern, std::move(open_function), kLoadAllShards) {
108 TensorSliceReader::TensorSliceReader(const string& filepattern,
111 : filepattern_(filepattern), open_function_(std::move(open_function)) {
112 VLOG(1) << "TensorSliceReader for " << filepattern;
113 Status s = Env::Default()->GetMatchingPaths(filepattern, &fnames_);
118 filepattern, ": ", s.ToString());
125 filepattern);
[all...] |
| tensor_slice_reader.h | 65 TensorSliceReader(const string& filepattern);
66 TensorSliceReader(const string& filepattern, OpenTableFunction open_function);
67 TensorSliceReader(const string& filepattern, OpenTableFunction open_function,
72 const string& filepattern() const { return filepattern_; } function in class:tensorflow::checkpoint::TensorSliceReader
|
| tensor_slice_reader_test.cc | 114 const string filepattern = strings::StrCat(fname_base, "_*"); local
115 TensorSliceReader reader(filepattern, std::move(open_function));
239 const string filepattern = strings::StrCat(fname_base, "_*"); local
240 TensorSliceReader reader(filepattern, std::move(open_function));
372 const string filepattern = strings::StrCat(fname_base, "_*"); local
374 filepattern, open_function, TensorSliceReader::kLoadAllShards);
390 filepattern, open_function, TensorSliceReader::kLoadAllShards);
|
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| checkpoint_utils.py | 40 def _get_checkpoint_filename(filepattern):
41 """Returns checkpoint filename given directory or specific filepattern."""
42 if gfile.IsDirectory(filepattern):
43 return saver.latest_checkpoint(filepattern)
44 return filepattern
47 def load_checkpoint(filepattern):
51 filepattern: Directory with checkpoints file or path to checkpoint.
59 filename = _get_checkpoint_filename(filepattern)
62 "given directory %s" % filepattern)
219 filepattern = _get_checkpoint_filename(checkpoint_dir
[all...] |
| /external/tensorflow/tensorflow/c/ |
| checkpoint_reader.h | 42 CheckpointReader(const string& filepattern, TF_Status* out_status);
|
| /external/tensorflow/tensorflow/python/util/ |
| py_checkpoint_reader.i | 162 def NewCheckpointReader(filepattern):
166 return CheckpointReader(compat.as_bytes(filepattern), status)
|
| /external/tensorflow/tensorflow/python/training/ |
| basic_session_run_hooks_test.py | [all...] |
