Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
Dbeam_search_decoder_test.py78 depth_ndims=0, argument
104 for _ in range(depth_ndims):
114 if depth_ndims > 0:
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)