Searched defs:last_batch_size (Results 1 – 2 of 2) sorted by relevance
937 int64_t last_batch_size = batch_sizes[0]; in operator ()() local996 int64_t last_batch_size = batch_sizes[num_steps - 1]; in operator ()() local
2978 def update_hidden_for_packed(cur_hidden, last_batch_size, batch_size, hiddens): argument2985 cur_hidden, last_batch_size, batch_size, inp_hidden argument