Home
last modified time | relevance | path

Searched refs:filepattern (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dtensor_slice_reader_cache.cc33 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 Dtensor_slice_reader.cc100 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 Dtensor_slice_reader_test.cc121 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 Dtensor_slice_reader.h65 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 Dtensor_slice_reader_cache.h43 const string& filepattern,
62 const string& filepattern,
/aosp_15_r20/external/tensorflow/third_party/gpus/
H A Dfind_cuda_config.py197 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 Dpy_checkpoint_reader.py82 def NewCheckpointReader(filepattern): argument
92 return CheckpointReader(compat.as_bytes(filepattern))
H A Dbasic_session_run_hooks_test.py1511 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 Dtensorflow.train.pbtxt261 argspec: "args=[\'filepattern\'], varargs=None, keywords=None, defaults=None"