Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorTransformerFunctions.cpp251 c10::IntArrayRef max_lengths, in _jagged_to_padded_dense_forward_cpu()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cu1417 c10::IntArrayRef max_lengths, in _fbgemm_jagged_to_padded_dense_forward()