Searched defs:max_seqlen (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/nested/_internal/ |
| H A 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
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | FunctionalInverses.cpp | 306 …4_t ragged_idx, const std::optional<Tensor>& min_seqlen, const std::optional<Tensor>& max_seqlen) { in _nested_view_from_jagged_inverse()
321 auto max_seqlen = at::_nested_get_max_seqlen(base); in _nested_get_values_inverse() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/ |
| H A D | NestedTensorTransformerUtils.cpp | 56 int64_t max_seqlen = -1; in cumulative_and_max_seq_len_nnz() local
|