Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc619 BundleEntryProto to_merge_entry; in MergeOneBundle() local
633 ParseEntryProto(iter->key(), iter->value(), &to_merge_entry)); in MergeOneBundle()
639 if (to_merge_entry.slices().empty()) { in MergeOneBundle()
645 for (int i = 0; i < to_merge_entry.slices_size(); ++i) { in MergeOneBundle()
647 *slot = to_merge_entry.slices(i); in MergeOneBundle()
649 CHECK_EQ(existing_entry.dtype(), to_merge_entry.dtype()); in MergeOneBundle()
651 TensorShape(to_merge_entry.shape())); in MergeOneBundle()
657 {DataFilename(prefix, to_merge_entry.shard_id(), num_shards), in MergeOneBundle()
659 to_merge_entry.set_shard_id(result.first->second); in MergeOneBundle()
660 merge_state->entries[key] = to_merge_entry; in MergeOneBundle()