Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc561 std::unordered_map<string, int32> shard_ids; member
656 auto result = merge_state->shard_ids.insert( in MergeOneBundle()
658 merge_state->shard_ids.size()}); in MergeOneBundle()
677 for (const auto& p : merge.shard_ids) { in MergeBundles()
679 << DataFilename(merged_prefix, p.second, merge.shard_ids.size()); in MergeBundles()
682 DataFilename(merged_prefix, p.second, merge.shard_ids.size()))); in MergeBundles()