Home
last modified time | relevance | path

Searched refs:shard_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/lib/io/
Dcache.cc353 LRUCache shard_[kNumShards]; member in tensorflow::table::__anon4220aee90111::ShardedLRUCache
367 shard_[s].SetCapacity(per_shard); in ShardedLRUCache()
374 return shard_[Shard(hash)].Insert(key, hash, value, charge, deleter); in Insert()
378 return shard_[Shard(hash)].Lookup(key, hash); in Lookup()
382 shard_[Shard(h->hash)].Release(handle); in Release()
386 shard_[Shard(hash)].Erase(key, hash); in Erase()
397 shard_[s].Prune(); in Prune()
403 total += shard_[s].TotalCharge(); in TotalCharge()
/external/tensorflow/tensorflow/core/kernels/
Drange_sampler.cc243 shard_(shard) { in FixedUnigramSampler()
259 shard_(shard) { in FixedUnigramSampler()
280 if (word_id % num_shards_ == shard_) weights_.push_back(0.0); in FillReservedIds()
298 if (word_id % num_shards_ == shard_) { in LoadFromFile()
318 if (word_id % num_shards_ == shard_) { in LoadFromUnigrams()
Drange_sampler.h239 int32 shard_; variable