Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
Dbeam_search_decoder_test.py77 def _test_gather_tree_from_array(self, member in TestGatherTree
127 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)