Home
last modified time | relevance | path

Searched defs:current_seq_len (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/nested/cuda/
DNestedTensorTransformerUtils.cpp25 int64_t current_seq_len = sizes_ptr[(i * size_tensor_stride)]; in get_nnz() local
60 auto current_seq_len = sizes_ptr[(i * size_tensor_stride)]; in cumulative_and_max_seq_len_nnz() local