Searched refs:_test_gather_tree_from_array (Results 1 – 1 of 1) sorted by relevance
77 def _test_gather_tree_from_array(self, member in TestGatherTree127 self._test_gather_tree_from_array()130 self._test_gather_tree_from_array(depth_ndims=1)133 self._test_gather_tree_from_array(depth_ndims=1, merged_batch_beam=True)136 self._test_gather_tree_from_array(depth_ndims=2)