Searched refs:merge_state (Results 1 – 25 of 44) sorted by relevance
12
595 MergeState* merge_state) { in MergeOneBundle() argument621 merge_state->num_shards += header.num_shards(); in MergeOneBundle()622 if (!merge_state->seen_first_bundle) { in MergeOneBundle()623 merge_state->seen_first_bundle = true; in MergeOneBundle()624 merge_state->endianness = header.endianness(); in MergeOneBundle()625 merge_state->version = header.version(); in MergeOneBundle()628 if (merge_state->endianness != header.endianness()) { in MergeOneBundle()635 merge_state->version.SerializeToString(&merge_version); in MergeOneBundle()650 const auto entry_iter = merge_state->entries.find(key); in MergeOneBundle()653 if (entry_iter != merge_state->entries.end() && in MergeOneBundle()[all …]
99 def merge_state(self, layers): member in PreprocessingLayer300 def merge_state(self, layers): member in CombinerPreprocessingLayer
303 norm_1.merge_state([norm_2, norm_3])
207 def merge_state(self, layers): member in Normalization
254 name: "merge_state"
252 name: "merge_state"
257 name: "merge_state"
251 name: "merge_state"