Searched defs:batch_sizes_t (Results 1 – 1 of 1) sorted by relevance
63 at::Tensor batch_sizes_t = at::empty(lengths[0], _lengths.options()); in _pack_padded_sequence() local122 auto batch_sizes_t = _batch_sizes.contiguous(); in _pack_padded_sequence_backward_symint() local143 auto batch_sizes_t = _batch_sizes.contiguous(); in _pad_packed_sequence() local