Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cu969 int dense_col = row - offset; in jagged_dense_dense_elementwise_jagged_output_opt_search_kernel_() local
1085 int dense_col = cols[real_row]; in jagged_dense_dense_elementwise_jagged_output_opt_gather_kernel_() local