| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A 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 …]
|
| H A D | tensor_slice_reader.cc | 100 TensorSliceReader::TensorSliceReader(const string& filepattern) in TensorSliceReader() argument
101 : TensorSliceReader(filepattern, OpenTableTensorSliceReader, in TensorSliceReader()
104 TensorSliceReader::TensorSliceReader(const string& filepattern, in TensorSliceReader() argument
106 : TensorSliceReader(filepattern, std::move(open_function), kLoadAllShards) { in TensorSliceReader()
109 TensorSliceReader::TensorSliceReader(const string& filepattern, in TensorSliceReader() argument
112 : filepattern_(filepattern), open_function_(std::move(open_function)) { in TensorSliceReader()
113 VLOG(1) << "TensorSliceReader for " << filepattern; in TensorSliceReader()
114 Status s = Env::Default()->GetMatchingPaths(filepattern, &fnames_); in TensorSliceReader()
119 filepattern, ": ", s.ToString()); in TensorSliceReader()
126 filepattern); in TensorSliceReader()
|
| H A D | tensor_slice_reader_test.cc | 121 const string filepattern = strings::StrCat(fname_base, "_*"); in SimpleFloatHelper() local
122 TensorSliceReader reader(filepattern, std::move(open_function)); in SimpleFloatHelper()
246 const string filepattern = strings::StrCat(fname_base, "_*"); in SimpleIntXHelper() local
247 TensorSliceReader reader(filepattern, std::move(open_function)); in SimpleIntXHelper()
550 const string filepattern = strings::StrCat(fname_base, "_*"); in CachedTensorSliceReaderTesterHelper() local
552 filepattern, open_function, TensorSliceReader::kLoadAllShards); in CachedTensorSliceReaderTesterHelper()
568 filepattern, open_function, TensorSliceReader::kLoadAllShards); in CachedTensorSliceReaderTesterHelper()
|
| H A 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
|
| H A D | tensor_slice_reader_cache.h | 43 const string& filepattern,
62 const string& filepattern,
|
| /aosp_15_r20/external/tensorflow/third_party/gpus/ |
| H A 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 …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | py_checkpoint_reader.py | 82 def NewCheckpointReader(filepattern): argument
92 return CheckpointReader(compat.as_bytes(filepattern))
|
| H A D | basic_session_run_hooks_test.py | 1511 self.filepattern = os.path.join(self.output_dir, 'timeline-*.json')
1521 return len(gfile.Glob(self.filepattern))
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.train.pbtxt | 261 argspec: "args=[\'filepattern\'], varargs=None, keywords=None, defaults=None"
|