Searched refs:_collect_previous_mask (Results 1 – 1 of 1) sorted by relevance
222 def _collect_previous_mask(x): function225 return nest.map_structure(_collect_previous_mask, input_tensors)