Beispiel #1
0
 def _test_body():
     if tf.executing_eagerly():
         beam_search_decoder._check_batch_beam(t, batch_size, beam_width)
     else:
         with self.cached_session():
             check_op = beam_search_decoder._check_batch_beam(
                 t, batch_size, beam_width
             )
             self.evaluate(check_op)
Beispiel #2
0
 def _test_body():
     # pylint: disable=protected-access
     if tf.executing_eagerly():
         beam_search_decoder._check_batch_beam(t, batch_size,
                                               beam_width)
     else:
         with self.cached_session():
             check_op = beam_search_decoder._check_batch_beam(
                 t, batch_size, beam_width)
             self.evaluate(check_op)
 def _test_body(t):
     beam_search_decoder._check_batch_beam(t, batch_size, beam_width)