| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_reader_cache.cc | 34 const string& filepattern, in GetReader() argument
41 return cache_->GetReader(filepattern, std::move(open_function), in GetReader()
54 const string& filepattern, in GetReader() argument
77 while (still_opening_.find(filepattern) != still_opening_.end()) { in GetReader()
82 if (readers_.find(filepattern) == readers_.end()) { in GetReader()
83 VLOG(1) << "Creating new TensorSliceReader for " << filepattern; in GetReader()
84 still_opening_.insert(filepattern); in GetReader()
89 new TensorSliceReader(filepattern, open_function, preferred_shard)); in GetReader()
94 readers_[filepattern] = std::make_pair(*func_ptr, reader); in GetReader()
98 CHECK_EQ(size_t{1}, still_opening_.erase(filepattern)); in GetReader()
[all …]
|
| D | tensor_slice_reader.cc | 99 TensorSliceReader::TensorSliceReader(const string& filepattern) in TensorSliceReader() argument
100 : TensorSliceReader(filepattern, OpenTableTensorSliceReader, in TensorSliceReader()
103 TensorSliceReader::TensorSliceReader(const string& filepattern, in TensorSliceReader() argument
105 : TensorSliceReader(filepattern, std::move(open_function), kLoadAllShards) { in TensorSliceReader()
108 TensorSliceReader::TensorSliceReader(const string& filepattern, in TensorSliceReader() argument
111 : filepattern_(filepattern), open_function_(std::move(open_function)) { in TensorSliceReader()
112 VLOG(1) << "TensorSliceReader for " << filepattern; in TensorSliceReader()
113 Status s = Env::Default()->GetMatchingPaths(filepattern, &fnames_); in TensorSliceReader()
118 filepattern, ": ", s.ToString()); in TensorSliceReader()
125 filepattern); in TensorSliceReader()
|
| D | tensor_slice_reader_test.cc | 115 const string filepattern = strings::StrCat(fname_base, "_*"); in SimpleFloatHelper() local
116 TensorSliceReader reader(filepattern, std::move(open_function)); in SimpleFloatHelper()
240 const string filepattern = strings::StrCat(fname_base, "_*"); in SimpleIntXHelper() local
241 TensorSliceReader reader(filepattern, std::move(open_function)); in SimpleIntXHelper()
373 const string filepattern = strings::StrCat(fname_base, "_*"); in TEST_SIMPLE_INT() local
375 filepattern, open_function, TensorSliceReader::kLoadAllShards); in TEST_SIMPLE_INT()
391 filepattern, open_function, TensorSliceReader::kLoadAllShards); in TEST_SIMPLE_INT()
|
| D | 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_; } in filepattern() function
|
| D | tensor_slice_reader_cache.h | 43 const string& filepattern,
62 const string& filepattern,
|
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| D | checkpoint_utils.py | 40 def _get_checkpoint_filename(filepattern): argument
42 if gfile.IsDirectory(filepattern):
43 return checkpoint_management.latest_checkpoint(filepattern)
44 return filepattern
47 def load_checkpoint(filepattern): argument
59 filename = _get_checkpoint_filename(filepattern)
62 "given directory %s" % filepattern)
219 filepattern = _get_checkpoint_filename(checkpoint_dir)
257 _set_variable_or_list_initializer(var, filepattern, tensor_name_in_ckpt)
297 _set_variable_or_list_initializer(var, filepattern, full_tensor_name)
|
| /external/tensorflow/tensorflow/python/util/ |
| D | py_checkpoint_reader.i | 162 def NewCheckpointReader(filepattern):
166 return CheckpointReader(compat.as_bytes(filepattern), status)
|
| /external/tensorflow/tensorflow/c/ |
| D | checkpoint_reader.h | 42 CheckpointReader(const string& filepattern, TF_Status* out_status);
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.pbtxt | 257 argspec: "args=[\'filepattern\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/python/training/ |
| D | basic_session_run_hooks_test.py | 1476 self.filepattern = os.path.join(self.output_dir, 'timeline-*.json')
1486 return len(gfile.Glob(self.filepattern))
|