Searched refs:block_map_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/platform/cloud/ |
| D | ram_file_block_cache.cc | 36 auto entry = block_map_.find(key); in Lookup()
37 if (entry != block_map_.end()) { in Lookup()
57 block_map_.emplace(std::make_pair(key, new_entry)); in Lookup()
64 RemoveBlock(block_map_.find(lru_list_.back())); in Trim()
88 auto fcmp = block_map_.upper_bound(fmax); in UpdateLRU()
89 if (fcmp != block_map_.begin() && key < (--fcmp)->first) { in UpdateLRU()
257 auto it = block_map_.find(lra_list_.back()); in Prune()
271 block_map_.clear(); in Flush()
284 auto it = block_map_.lower_bound(begin); in RemoveFile_Locked()
285 while (it != block_map_.end() && it->first.first == filename) { in RemoveFile_Locked()
[all …]
|
| D | ram_file_block_cache.h | 228 BlockMap block_map_ TF_GUARDED_BY(mu_);
|
| /external/tensorflow/tensorflow/c/experimental/filesystem/plugins/gcs/ |
| D | ram_file_block_cache.cc | 39 auto entry = block_map_.find(key); in Lookup()
40 if (entry != block_map_.end()) { in Lookup()
57 block_map_.emplace(std::make_pair(key, new_entry)); in Lookup()
64 RemoveBlock(block_map_.find(lru_list_.back())); in Trim()
89 auto fcmp = block_map_.upper_bound(fmax); in UpdateLRU()
90 if (fcmp != block_map_.begin() && key < (--fcmp)->first) { in UpdateLRU()
264 auto it = block_map_.find(lra_list_.back()); in Prune()
278 block_map_.clear(); in Flush()
291 auto it = block_map_.lower_bound(begin); in RemoveFile_Locked()
292 while (it != block_map_.end() && it->first.first == filename) { in RemoveFile_Locked()
[all …]
|
| D | ram_file_block_cache.h | 244 BlockMap block_map_ ABSL_GUARDED_BY(mu_);
|
| /external/ruy/ruy/ |
| D | trmul.cc | 64 block_map_(block_map), in TrMulTask()
81 const int size = NumBlocksPerSide(side, block_map_); in Run()
88 const int num_blocks = NumBlocks(block_map_); in Run()
101 GetBlockByIndex(block_map_, block_id, &block); in Run()
104 GetBlockMatrixCoords(block_map_, block, &start, &end); in Run()
220 if (runahead_block >= NumBlocksPerSide(runahead_side, block_map_)) { in EnsurePacked()
224 GetBlockMatrixCoords(runahead_side, block_map_, runahead_block, in EnsurePacked()
235 const BlockMap& block_map_; member in ruy::__anonaab2759e0111::TrMulTask
|
| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
| D | format_converter.cc | 60 block_map_(block_map) { in FormatConverter()
85 int num_block_dims = block_map_.size(); in DenseToSparse()
107 int mapped_dim = block_map_[i]; in DenseToSparse()
225 block_map_ = TfLiteIntArrayToVector(sparsity.block_map); in FormatConverter()
245 block_size_.resize(block_map_.size()); in FormatConverter()
247 if (block_dim < block_map_.size() && block_map_[block_dim] == i) { in FormatConverter()
274 const int orig_dim = block_map_[block_idx]; in Populate()
|
| D | format_converter.h | 96 std::vector<int> block_map_; variable
|