Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cu967 int dense_row = first; in jagged_dense_dense_elementwise_jagged_output_opt_search_kernel_() local
1084 int dense_row = rows[real_row]; in jagged_dense_dense_elementwise_jagged_output_opt_gather_kernel_() local