Searched refs:merged_batch_beam (Results 1 – 1 of 1) sorted by relevance
79 merged_batch_beam=False): argument108 if merged_batch_beam:133 self._test_gather_tree_from_array(depth_ndims=1, merged_batch_beam=True)