Searched defs:min_seqlen (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/nested/_internal/ |
| D | nested_tensor.py | 524 values, offsets, ragged_idx=1, min_seqlen=None, max_seqlen=None argument
546 values, offsets, lengths, ragged_idx=1, min_seqlen=None, max_seqlen=None argument
|
| /external/pytorch/aten/src/ATen/ |
| D | FunctionalInverses.cpp | 306 …onst std::optional<Tensor>& lengths, int64_t ragged_idx, const std::optional<Tensor>& min_seqlen, … in _nested_view_from_jagged_inverse()
320 auto min_seqlen = at::_nested_get_min_seqlen(base); in _nested_get_values_inverse() local
|