| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_reader_cache.cc | 33 const string& filepattern, in GetReader() argument
40 return cache_->GetReader(filepattern, std::move(open_function), in GetReader()
53 const string& filepattern, in GetReader() argument
76 while (still_opening_.find(filepattern) != still_opening_.end()) { in GetReader()
81 if (readers_.find(filepattern) == readers_.end()) { in GetReader()
82 VLOG(1) << "Creating new TensorSliceReader for " << filepattern; in GetReader()
83 still_opening_.insert(filepattern); in GetReader()
88 new TensorSliceReader(filepattern, open_function, preferred_shard)); in GetReader()
93 readers_[filepattern] = std::make_pair(*func_ptr, reader); in GetReader()
97 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/third_party/gpus/ |
| D | find_cuda_config.py | 197 def _not_found_error(base_paths, relative_paths, filepattern): argument
202 (filepattern, relative_paths, base_paths))
205 def _find_file(base_paths, relative_paths, filepattern): argument
207 for file in glob.glob(os.path.join(path, filepattern)):
209 raise _not_found_error(base_paths, relative_paths, filepattern)
215 filepattern = library_name + ".lib"
217 filepattern = "%s*.dylib" % (".".join(["lib" + library_name] +
220 filepattern = ".".join(["lib" + library_name, "so"] +
222 return _find_file(base_paths, _library_paths(), filepattern)
231 for filepattern in filepatterns:
[all …]
|
| /external/tensorflow/tensorflow/python/training/ |
| D | py_checkpoint_reader.py | 85 def NewCheckpointReader(filepattern): argument
95 return CheckpointReader(compat.as_bytes(filepattern))
|
| D | basic_session_run_hooks_test.py | 1515 self.filepattern = os.path.join(self.output_dir, 'timeline-*.json')
1525 return len(gfile.Glob(self.filepattern))
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.pbtxt | 261 argspec: "args=[\'filepattern\'], varargs=None, keywords=None, defaults=None"
|