Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc567 MergeState* merge_state) { in MergeOneBundle() argument
593 merge_state->num_shards += header.num_shards(); in MergeOneBundle()
594 if (!merge_state->seen_first_bundle) { in MergeOneBundle()
595 merge_state->seen_first_bundle = true; in MergeOneBundle()
596 merge_state->endianness = header.endianness(); in MergeOneBundle()
597 merge_state->version = header.version(); in MergeOneBundle()
600 if (merge_state->endianness != header.endianness()) { in MergeOneBundle()
607 merge_state->version.SerializeToString(&merge_version); in MergeOneBundle()
622 const auto entry_iter = merge_state->entries.find(key); in MergeOneBundle()
625 if (entry_iter != merge_state->entries.end() && in MergeOneBundle()
[all …]