Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cu1217 int used_shared_kb = round_down(shared_kb * 2 / 3, 16); in jagged_dense_dense_elementwise_jagged_output_matches_opt() local
1221 int used_shared_kb = shared_kb; in jagged_dense_dense_elementwise_jagged_output_matches_opt() local
1281 int used_shared_kb = round_down(shared_kb * 2 / 3, 16); in calc_used_shared_bytes() local
1285 int used_shared_kb = shared_kb; in calc_used_shared_bytes() local