Lines Matching refs:range_size
589 range_size=beam_width * vocab_size,
608 range_size=beam_width,
625 range_size=beam_width,
639 range_size=beam_width,
739 range_size, gather_shape):
750 range_size: The number of values in each range. Likely equal to beam_width.
768 range_size=range_size,
777 range_size,
790 range_size: The number of values in each range. Likely equal to beam_width.
804 range_ = array_ops.expand_dims(math_ops.range(batch_size) * range_size, 1)
