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)
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)