Searched refs:_nested_get_max_seqlen (Results 1 – 3 of 3) sorted by relevance
321 auto max_seqlen = at::_nested_get_max_seqlen(base); in _nested_get_values_inverse()
1628 @register_jagged_func(torch.ops.aten._nested_get_max_seqlen.default, "self: jt_all")1629 def _nested_get_max_seqlen(func, *args, **kwargs): function
450 aten::_nested_get_max_seqlen